Clean up frames whether error occured or not upon __exit__

matplotloom/loom.py

@@ -53,18 +53,19 @@ def __enter__(self):
         return self
 
     def __exit__(self, exc_type, exc_value, traceback):
-        if exc_type is None:
-            self.save_video()
-        else:
+        try:
+            if exc_type is None:
+                self.save_video()
+            else:
+                if self.verbose:
+                    print(f"An error occurred: {exc_type.__name__}: {exc_value}")
+                    print("Animation was not saved.")
+        finally:
             if not self.keep_frames:
                 for frame_filepath in self.frame_filepaths:
                     if frame_filepath.exists():
                         frame_filepath.unlink()
-
-            if self.verbose:
-                print(f"An error occurred: {exc_type.__name__}: {exc_value}")
-                print("Animation was not saved.")
-
+
         return False  # Propagate the exception if there was one
 
     def save_frame(self, fig, frame_number=None):