flush only when saving

logix-project · Nov 24, 2023 · 1a3f941 · 1a3f941
1 parent 05e37ca
commit 1a3f941
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/analog/analog.py b/analog/analog.py
@@ -237,7 +237,8 @@ def __exit__(self, exc_type, exc_value, traceback) -> None:
         interfere with further operations on the model or with future logging sessions.
         """
         self.hessian_handler.on_exit(self.logging_handler.current_log)
-        self.storage_handler.flush()
+        if self.save:
+            self.storage_handler.flush()
         self.logging_handler.clear()
 
         self.reset()