Fix mlxtend version

DIAGNijmegen · Jan 12, 2023 · 3e69f54 · 3e69f54
1 parent fea260e
commit 3e69f54
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/picai_baseline/nnunet/training_docker/Dockerfile b/src/picai_baseline/nnunet/training_docker/Dockerfile
@@ -33,7 +33,8 @@ RUN SITE_PKG=`pip3 show nnunet | grep "Location:" | awk '{print $2}'` && \
 RUN pip3 install \
     picai_eval>=1.4.4 \
     picai_prep>=2.1.2 \
-    picai_baseline>=0.7.0
+    picai_baseline>=0.7.0 \
+    mlxtend==0.19.0
 
 # copy custom files to nnUNet folder
 COPY nnUNetTrainerV2_focalLoss.py /tmp/nnUNetTrainerV2_focalLoss.py