Update src/imgtools/io/writers/abstract_base_writer.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
bhklab · Jan 17, 2025 · 12137eb · 12137eb
1 parent 32209e8
commit 12137eb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/imgtools/io/writers/abstract_base_writer.py b/src/imgtools/io/writers/abstract_base_writer.py
@@ -292,7 +292,9 @@ def __exit__(
         logger.debug(f"Exiting context manager for {self.__class__.__name__}")
         if self.index_file.exists():
             logger.debug(f"Removing lock file {self._get_index_lock()}")
-            self._get_index_lock().unlink()
+            lock_file = self._get_index_lock()
+            if lock_file.exists():
+                lock_file.unlink()
 
         # if the root directory is empty, aka we created it but didn't write anything, delete it
         if (