From 8fff656049bfbc15655943a7154b5c99623637e8 Mon Sep 17 00:00:00 2001
From: Alessio Buccino <alejoe9187@gmail.com>
Date: Tue, 27 Aug 2024 10:26:36 +0200
Subject: [PATCH] Update KS4 to 4.0.16

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

diff --git a/kilosort4/Dockerfile b/kilosort4/Dockerfile
index 5306ea6..e139b3a 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.12
+RUN pip install kilosort==4.0.16
diff --git a/kilosort4/build.sh b/kilosort4/build.sh
index 79095a1..8f7e3ca 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.12_cuda-12.0.0 .
+docker build -t spikeinterface/kilosort4-base:latest -t spikeinterface/kilosort4-base:4.0.16_cuda-12.0.0 .