From 0113ed36b6f9d773cf4755d7db0b11b3cf606280 Mon Sep 17 00:00:00 2001
From: Ads Dawson <104169244+GangGreenTemperTatum@users.noreply.github.com>
Date: Fri, 20 Jun 2025 06:51:02 -0400
Subject: [PATCH] chore: pin deps in dockerfile for future version updates

---
 airtbench/container/Dockerfile | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/airtbench/container/Dockerfile b/airtbench/container/Dockerfile
index 5194202..f06227d 100644
--- a/airtbench/container/Dockerfile
+++ b/airtbench/container/Dockerfile
@@ -1,12 +1,20 @@
-FROM jupyter/scipy-notebook
+FROM jupyter/scipy-notebook@sha256:fca4bcc9cbd49d9a15e0e4df6c666adf17776c950da9fa94a4f0a045d5c4ad33
 
 RUN pip install \
-    torch \
-    torchvision \
-    torchaudio \
-    catboost \
-    GPy \
-    lightgbm \
-    xgboost \
-    kornia \
-    lief
\ No newline at end of file
+    adversarial-robustness-toolbox==1.19.1 \
+    catboost==1.2.8 \
+    eagerpy==0.30.0 \
+    foolbox==3.3.4 \
+    GPy==1.13.2 \
+    kornia==0.8.1 \
+    lief==0.16.5 \
+    lightgbm==4.6.0 \
+    numpy==1.24.3 \
+    plotly==6.0.1 \
+    requests==2.31.0 \
+    torch==2.7.0 \
+    torchaudio==2.7.0 \
+    torchvision==0.22.0 \
+    xgboost==3.0.0
+
+RUN pip install kornia_rs==0.1.9
\ No newline at end of file