From 334d5738575023698898c3e8cdffa8eee0ca656d Mon Sep 17 00:00:00 2001 From: silvandeleemput Date: Thu, 23 Nov 2023 21:42:08 +0100 Subject: [PATCH] update mhub model definition import Dockerfile --- models/gc_nnunet_pancreas/dockerfiles/Dockerfile | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/models/gc_nnunet_pancreas/dockerfiles/Dockerfile b/models/gc_nnunet_pancreas/dockerfiles/Dockerfile index 3fe9a951..9f6b56a5 100644 --- a/models/gc_nnunet_pancreas/dockerfiles/Dockerfile +++ b/models/gc_nnunet_pancreas/dockerfiles/Dockerfile @@ -31,12 +31,9 @@ RUN pip3 install --no-cache-dir -r /opt/algorithm/requirements.txt RUN SITE_PKG=`pip3 show nnunet | grep "Location:" | awk '{print $2}'` && \ mv /opt/algorithm/nnUNetTrainerV2_Loss_CE_checkpoints.py "$SITE_PKG/nnunet/training/network_training/nnUNetTrainerV2_Loss_CE_checkpoints.py" -# Clone the main branch of MHubAI/models TODO check if ok -RUN git stash \ - && git fetch https://github.com/MHubAI/models.git main \ - && git merge FETCH_HEAD \ - && git sparse-checkout set "models/gc_nnunet_pancreas" \ - && git fetch https://github.com/MHubAI/models.git main +# Import the MHub model definiton +ARG MHUB_MODELS_REPO +RUN buildutils/import_mhub_model.sh gc_nnunet_pancreas ${MHUB_MODELS_REPO} # Add algorithm files to python path ENV PYTHONPATH=/opt/algorithm:/app