wip

Author: alejoe91

pykilosort/Dockerfile

@@ -30,14 +30,16 @@ RUN echo ". $CONDA_DIR/etc/profile.d/conda.sh" >> /root/.profile
 RUN conda init bash
 
 # Install python port of pykilosort
-RUN git clone -b drift_test_stable https://github.com/kushbanga/pykilosort.git /src/pykilosort
+RUN git clone https://github.com/int-brain-lab/pykilosort.git /src/pykilosort
 WORKDIR /src/pykilosort
 # Switch to MouseLand/develop branch - may revert to kushbanga/drift_test_stable in the future
-RUN git remote add MouseLand https://github.com/MouseLand/pykilosort.git && \
-    git fetch MouseLand && git checkout MouseLand/develop && git checkout -b develop
+# RUN git remote add MouseLand https://github.com/MouseLand/pykilosort.git && \
+#     git fetch MouseLand && git checkout MouseLand/develop && git checkout -b develop
 
 # modify env file so that env extends on base
 RUN sed -i "s/pyks2/base/" pyks2.yml
+RUN sed -i "s/- spikeinterface//" pyks2.yml
+RUN cat pyks2.yml
 
 # Create environment
 RUN conda env update --name base --file pyks2.yml --prune

pykilosort/build.sh

@@ -1,3 +1,3 @@
 #!/bin/bash
 
-docker build -t spikeinterface/pykilosort-base:latest -t spikeinterface/pykilosort-base:0.1.0 .
+docker build -t spikeinterface/pykilosort-base:latest -t spikeinterface/pykilosort-base:ibl-1.0.2 .

tests/test_containers_gpu.py

@@ -8,6 +8,22 @@
 
 os.environ['SINGULARITY_DISABLE_CACHE'] = 'true'
 
+DOCKER_SINGULARITY = "docker" # "singularity"
+
+def _run_kwargs():
+    test_recording, _ = se.toy_example(
+        duration=30,
+        seed=0,
+        num_channels=64,
+        num_segments=1
+    )
+    test_recording = test_recording.save(name='toy')
+    kwargs = dict(recording=test_recording, verbose=True)
+    if DOCKER_SINGULARITY == "singularity":
+        kwargs["singularity_image"] = True
+    else:
+        kwargs["docker_image"] = True
+    return kwargs
 
 @pytest.fixture(autouse=True)
 def work_dir(request, tmp_path):
@@ -34,7 +50,12 @@ def run_kwargs(work_dir):
         num_segments=1
     )
     test_recording = test_recording.save(name='toy')
-    return dict(recording=test_recording, verbose=True, singularity_image=True)
+    kwargs = dict(recording=test_recording, verbose=True)
+    if DOCKER_SINGULARITY == "singularity":
+        kwargs["singularity_image"] = True
+    else:
+        kwargs["docker_image"] = True
+    return kwargs
 
 
 def test_kilosort2(run_kwargs):
@@ -61,3 +82,7 @@ def test_yass(run_kwargs):
 def test_pykilosort(run_kwargs):
     sorting = ss.run_pykilosort(output_folder="pykilosort", **run_kwargs)
     print(sorting)
+
+if __name__ == "__main__":
+    kwargs = _run_kwargs()
+    test_kilosort2(kwargs)

tests/test_singularity_containers.py

@@ -8,6 +8,8 @@
 
 os.environ['SINGULARITY_DISABLE_CACHE'] = 'true'
 
+DOCKER_SINGULARITY = "docker" # "singularity"
+
 
 @pytest.fixture(autouse=True)
 def work_dir(request, tmp_path):
@@ -34,7 +36,12 @@ def run_kwargs(work_dir):
         num_segments=1
     )
     test_recording = test_recording.save(name='toy')
-    return dict(recording=test_recording, verbose=True, singularity_image=True)
+    kwargs = dict(recording=test_recording, verbose=True)
+    if DOCKER_SINGULARITY == "singularity":
+        kwargs["singularity_image"] = True
+    else:
+        kwargs["docker_image"] = True
+    return kwargs
 
 
 def test_spykingcircus(run_kwargs):
@@ -75,3 +82,8 @@ def test_hdsort(run_kwargs):
 def test_kilosort1(run_kwargs):
     sorting = ss.run_kilosort(output_folder="kilosort", useGPU=False, **run_kwargs)
     print(sorting)
+
+
+if __name__ == "__main__":
+    kwargs = run_kwargs(None)
+    test_ironclust(kwargs)