making sure we use version

WIPACrepo · Aug 30, 2023 · 4769b4e · 4769b4e
1 parent d3e392c
commit 4769b4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet b/icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet
@@ -1,5 +1,5 @@
 ARG VERSION=current
-FROM icecube/icetray-ml:icetray-v1.7.2-cuda11.8-cudnn8-ubuntu22.04
+FROM icecube/icetray-ml:icetray-${VERSION}-cuda11.8-cudnn8-ubuntu22.04
 
 RUN pip3 install--upgrade pip
 RUN pip3 install --upgrade astropy