feat: save spike interface results with relative path

ttngu207 · ttngu207 · commit 67a1ffc76726 · 2024-04-17T09:15:31.000-05:00
diff --git a/element_array_ephys/spike_sorting/si_spike_sorting.py b/element_array_ephys/spike_sorting/si_spike_sorting.py
@@ -150,7 +150,7 @@ def make(self, key):
         # Run preprocessing and save results to output folder
         si_preproc_func = getattr(si_preprocessing, params["SI_PREPROCESSING_METHOD"])
         si_recording = si_preproc_func(si_recording)
-        si_recording.dump_to_pickle(file_path=recording_file)
+        si_recording.dump_to_pickle(file_path=recording_file, relative_to=output_dir)
 
         self.insert1(
             {
@@ -203,7 +203,7 @@ def make(self, key):
         sorting_save_path = (
             output_dir / sorter_name / "spike_sorting" / "si_sorting.pkl"
         )
-        si_sorting.dump_to_pickle(sorting_save_path)
+        si_sorting.dump_to_pickle(sorting_save_path, relative_to=output_dir)
 
         self.insert1(
             {

Original file line number	Diff line number	Diff line change
`@@ -150,7 +150,7 @@ def make(self, key):`
`150`	`150`	`# Run preprocessing and save results to output folder`
`151`	`151`	`si_preproc_func = getattr(si_preprocessing, params["SI_PREPROCESSING_METHOD"])`
`152`	`152`	`si_recording = si_preproc_func(si_recording)`
`153`		`- si_recording.dump_to_pickle(file_path=recording_file)`
	`153`	`+ si_recording.dump_to_pickle(file_path=recording_file, relative_to=output_dir)`
`154`	`154`
`155`	`155`	`self.insert1(`
`156`	`156`	`{`
`@@ -203,7 +203,7 @@ def make(self, key):`
`203`	`203`	`sorting_save_path = (`
`204`	`204`	`output_dir / sorter_name / "spike_sorting" / "si_sorting.pkl"`
`205`	`205`	`)`
`206`		`- si_sorting.dump_to_pickle(sorting_save_path)`
	`206`	`+ si_sorting.dump_to_pickle(sorting_save_path, relative_to=output_dir)`
`207`	`207`
`208`	`208`	`self.insert1(`
`209`	`209`	`{`