From eefd75866c945b6215799ff4e880361257097591 Mon Sep 17 00:00:00 2001
From: Alessio Buccino <alejoe9187@gmail.com>
Date: Fri, 18 Oct 2024 10:58:59 +0200
Subject: [PATCH] Update KS4 4.0.18

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

diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile
index e139b3a..3647f63 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.16
+RUN pip install kilosort==4.0.18
diff --git a/kilosort4/build.sh b/kilosort4/build.sh
index 8f7e3ca..32763ba 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.16_cuda-12.0.0 .
+docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.18_cuda-12.0.0 .