Dockerfile - fixed source pull to version tag

MHubAI · Apr 18, 2024 · 5cc1190 · 5cc1190
1 parent 777ff39
commit 5cc1190
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/models/gc_node21_baseline/dockerfiles/Dockerfile b/models/gc_node21_baseline/dockerfiles/Dockerfile
@@ -10,11 +10,10 @@ RUN pip3 install --no-cache-dir torch==2.0.1+cu118 torchvision==0.15.2+cu118 -f
 RUN apt update && apt install -y --no-install-recommends git-lfs && rm -rf /var/lib/apt/lists/*
 
 # Install node21 baseline algorithm
-#  - Git clone to tmp directory (main branch, commit 6e57f5c564eb1d527e0f030de9755179b213731a)
+#  - Git clone to tmp directory (main branch, tag v1.0addedtag)
 #  - Extract relevant files to /opt/algorithm/
 #  - Remove tmp directory
-RUN git clone https://github.com/node21challenge/node21_detection_baseline.git /tmp/algorithm && \
-    cd /tmp/algorithm && git reset --hard 6e57f5c564eb1d527e0f030de9755179b213731a && cd /app && \
+RUN git clone --depth 1 --branch v1.0addedtag https://github.com/node21challenge/node21_detection_baseline.git /tmp/algorithm && \
     mkdir -p /opt/algorithm && \
     mv /tmp/algorithm/process.py /opt/algorithm/process.py && \
     mv /tmp/algorithm/postprocessing.py /opt/algorithm/postprocessing.py && \