Fix classifier path

LorenFrankLab · Jan 14, 2024 · bae756d · bae756d
1 parent fe857b6
commit bae756d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/spyglass/decoding/v1/clusterless.py b/src/spyglass/decoding/v1/clusterless.py
@@ -262,7 +262,7 @@ def make(self, key):
         )
         key["results_path"] = results_path
 
-        classifier_path = results_path.strip(".nc") + ".pkl"
+        classifier_path = results_path.with_suffix(".pkl")
         classifier.save_model(classifier_path)
         key["classifier_path"] = classifier_path