From 8513a4d30e533baae18759325f7987a1d5b952a3 Mon Sep 17 00:00:00 2001
From: Alessio Buccino <alejoe9187@gmail.com>
Date: Tue, 30 Apr 2024 19:07:51 +0200
Subject: [PATCH] Update KS4 to v4.0.6

---
 kilosort4/Dockerfile | 2 +-
 kilosort4/build.sh   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile
index 001da23..5f64372 100644
--- a/kilosort4/Dockerfile
+++ b/kilosort4/Dockerfile
@@ -35,4 +35,4 @@ RUN conda init bash
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 # install kilosort4
-RUN pip install kilosort==4.0
+RUN pip install kilosort==4.0.6
diff --git a/kilosort4/build.sh b/kilosort4/build.sh
index 1c735c3..ebb5cae 100755
--- a/kilosort4/build.sh
+++ b/kilosort4/build.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 
-docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0_cuda-12.0.0 .
+docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.6_cuda-12.0.0 .