Skip to content

Commit 8dfc858

Browse files
committed
chore: 🎨 run docker if the package is not built into spikeinterface
1 parent d47be56 commit 8dfc858

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

element_array_ephys/spike_sorting/si_spike_sorting.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -205,23 +205,23 @@ def make(self, key):
205205
ephys.ClusteringTask * ephys.ClusteringParamSet & key
206206
).fetch1("clustering_method", "clustering_output_dir", "params")
207207
output_dir = find_full_path(ephys.get_ephys_root_data_dir(), output_dir)
208-
209-
# Get sorter method and create output directory.
210208
sorter_name = clustering_method.replace(".", "_")
211209
recording_file = output_dir / sorter_name / "recording" / "si_recording.pkl"
212210
si_recording: si.BaseRecording = si.load_extractor(recording_file)
213211

214212
# Run sorting
213+
# Sorting performed in a dedicated docker environment if the sorter is not built in the spikeinterface package.
215214
si_sorting: si.sorters.BaseSorter = si.sorters.run_sorter(
216215
sorter_name=sorter_name,
217216
recording=si_recording,
218217
output_folder=output_dir / sorter_name / "spike_sorting",
219218
remove_existing_folder=True,
220219
verbose=True,
221-
docker_image=True,
220+
docker_image=sorter_name not in si.sorters.installed_sorters(),
222221
**params.get("SI_SORTING_PARAMS", {}),
223222
)
224223

224+
# Save sorting object
225225
sorting_save_path = (
226226
output_dir / sorter_name / "spike_sorting" / "si_sorting.pkl"
227227
)

0 commit comments

Comments
 (0)