Remove temporary model files after failed writes

2026-10-27 16:53:49 +00:00 · 2026-10-27 16:53:49 +00:00 · f629b16cd2
commit f629b16cd2
parent fb31b3ed83
1 changed files with 21 additions and 4 deletions
--- a/great_ai/large_file/large_file/large_file_base.py
+++ b/great_ai/large_file/large_file/large_file_base.py
@ -146,11 +146,13 @@ class LargeFileBase(ABC):
    ) -> Literal[False]:
        self._file.close()

-        if type is None:
-            if "w" in self._mode:
-                self.push(Path(self._file.name))
+        if "w" in self._mode:
+            try:
+                if type is None:
+                    self.push(Path(self._file.name))
+            finally:
                os.unlink(self._file.name)
-        else:
+        elif type is not None:
            logger.exception("Could not finish operation.")

        return False