From 3e69f54a28db854c2fb81f883ec23bc34e428c08 Mon Sep 17 00:00:00 2001 From: Joeran Bosma Date: Thu, 12 Jan 2023 15:48:01 +0100 Subject: [PATCH] Fix mlxtend version --- src/picai_baseline/nnunet/training_docker/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/picai_baseline/nnunet/training_docker/Dockerfile b/src/picai_baseline/nnunet/training_docker/Dockerfile index 36249e5..47c26c8 100644 --- 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