File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
element_array_ephys/spike_sorting Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -205,23 +205,23 @@ def make(self, key):
205
205
ephys .ClusteringTask * ephys .ClusteringParamSet & key
206
206
).fetch1 ("clustering_method" , "clustering_output_dir" , "params" )
207
207
output_dir = find_full_path (ephys .get_ephys_root_data_dir (), output_dir )
208
-
209
- # Get sorter method and create output directory.
210
208
sorter_name = clustering_method .replace ("." , "_" )
211
209
recording_file = output_dir / sorter_name / "recording" / "si_recording.pkl"
212
210
si_recording : si .BaseRecording = si .load_extractor (recording_file )
213
211
214
212
# Run sorting
213
+ # Sorting performed in a dedicated docker environment if the sorter is not built in the spikeinterface package.
215
214
si_sorting : si .sorters .BaseSorter = si .sorters .run_sorter (
216
215
sorter_name = sorter_name ,
217
216
recording = si_recording ,
218
217
output_folder = output_dir / sorter_name / "spike_sorting" ,
219
218
remove_existing_folder = True ,
220
219
verbose = True ,
221
- docker_image = True ,
220
+ docker_image = sorter_name not in si . sorters . installed_sorters () ,
222
221
** params .get ("SI_SORTING_PARAMS" , {}),
223
222
)
224
223
224
+ # Save sorting object
225
225
sorting_save_path = (
226
226
output_dir / sorter_name / "spike_sorting" / "si_sorting.pkl"
227
227
)
You can’t perform that action at this time.
0 commit comments