fixes

Author: alejoe91

tests/test_containers_gpu.py

@@ -12,7 +12,7 @@
 DOCKER_SINGULARITY = "singularity" # "docker"
 
 
-def generate_run_kwargs(DOCKER_SINGULARITY):
+def generate_run_kwargs():
     test_recording, _ = se.toy_example(
         duration=30,
         seed=0,
@@ -75,5 +75,5 @@ def test_pykilosort(run_kwargs):
     print(sorting)
 
 if __name__ == "__main__":
-    kwargs = _run_kwargs()
+    kwargs = generate_run_kwargs()
     test_pykilosort(kwargs)

tests/test_singularity_containers.py

@@ -12,7 +12,7 @@
 DOCKER_SINGULARITY = "singularity" # "docker"
 
 
-def generate_run_kwargs(DOCKER_SINGULARITY):
+def generate_run_kwargs():
     test_recording, _ = se.toy_example(
         duration=30,
         seed=0,
@@ -99,5 +99,5 @@ def test_combinato(run_kwargs):
     print(sorting)
 
 if __name__ == "__main__":
-    kwargs = run_kwargs(None)
+    kwargs = generate_run_kwargs()
     test_ironclust(kwargs)