Update Dockerfile

casus · Sep 27, 2024 · 835731c · 835731c
1 parent b0bf71d
commit 835731c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -48,7 +48,8 @@ RUN micromamba create -n $ENV_NAME -y && \
     neptune==1.10.2 -y && \
     /usr/local/bin/micromamba run -n $ENV_NAME pip3 install opencv-python==4.9.0.80 \
     tensorflow-addons==0.23.0 \
-    cupy-cuda12x==13.3.0 && \
+    cupy-cuda12x==13.3.0 \
+    pytest && \
     /usr/local/bin/micromamba run -n $ENV_NAME python -m pip install .
 
 # Default command when the container starts (optional)