Update SORTER_DOCKER_MAP
@@ -11,25 +11,26 @@ from spikeinterface.core.core_tools import check_json, recursive_path_modifier,
from .sorterlist import sorter_dict
from .utils import SpikeSortingError, has_nvidia
+REGISTRY = 'spikeinterface'
+
+SORTER_DOCKER_MAP = dict(
+ klusta='klusta',
+ mountainsort4='mountainsort4',
+ spykingcircus='spyking-circus',
+ tridesclous='tridesclous',
+ # Matlab compiled sorters:
+ ironclust='ironclust-compiled',
+ kilosort='kilosort-compiled',
+ kilosort2='kilosort2-compiled',
+ kilosort3='kilosort3-compiled',
+ waveclus='waveclus-compiled',
+ hdsort='hdsort-compiled'
+)
+
SORTER_DOCKER_MAP = {
- name: f"spikeinterface/{name}-base" for name in [
- "tridesclous",
- "spyking-circus",
- "mountainsort4",
- "klusta",
- ]
+ k: f'{REGISTRY}/{v}-base'
+ for k, v in SORTER_DOCKER_MAP.items()
}
-SORTER_DOCKER_MAP.update(
- {
- name: f"spikeinterface/{name}-compiled-base" for name in [
- "ironclust",
- "kilosort",
- "kilosort2",
- "kilosort2_5",
- "kilosort3",
- ]
- }
-)
_common_param_doc = """
