Clean up saved model paths

LorenFrankLab · Jan 8, 2024 · d14f40f · d14f40f
1 parent 3995e24
commit d14f40f
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/spyglass/decoding/decoding_merge.py b/src/spyglass/decoding/decoding_merge.py
@@ -52,3 +52,18 @@ def cleanup(self):
             if str(path) not in table_results_paths:
                 logger.info(f"Removing {path}")
                 path.unlink()
+
+        table_model_paths = list(
+            chain(
+                *[
+                    part_parent_table.fetch("classifier_path").tolist()
+                    for part_parent_table in self.merge_get_parent(
+                        multi_source=True
+                    )
+                ]
+            )
+        )
+        for path in Path(config["SPYGLASS_ANALYSIS_DIR"]).glob("**/*.pkl"):
+            if str(path) not in table_model_paths:
+                logger.info(f"Removing {path}")
+                path.unlink()