fix to traindiff tests GH workflow

mlcommons · Feb 13, 2024 · 0447538 · 0447538
1 parent 802e709
commit 0447538
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/traindiffs_tests.yml b/.github/workflows/traindiffs_tests.yml
@@ -29,4 +29,4 @@ jobs:
       - name: Run containerized traindiffs test
         run: |
           docker pull us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_${{ github.head_ref || github.ref_name }}  
-          docker run -v $HOME/data/:/data/ -v $HOME/experiment_runs/:/experiment_runs -v $HOME/experiment_runs/logs:/logs --gpus all --ipc=host us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_${{ github.head_ref || github.ref_name }} algoperf_${{ github.head_ref || github.ref_name }} --traindiffs_test true
+          docker run -v $HOME/data/:/data/ -v $HOME/experiment_runs/:/experiment_runs -v $HOME/experiment_runs/logs:/logs --gpus all --ipc=host us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/algoperf_${{ github.head_ref || github.ref_name }} --traindiffs_test true
diff --git a/docker/scripts/startup.sh b/docker/scripts/startup.sh
@@ -54,10 +54,7 @@ SAVE_CHECKPOINTS="true"
 # Pass flag
 while [ "$1" != "" ]; do
     case $1 in
-	--traindiffs_test)
-	    shift
-            TEST=$1
-	    ;;
+
         -d | --dataset) 
             shift
             DATASET=$1
@@ -126,6 +123,10 @@ while [ "$1" != "" ]; do
             shift
             RNG_SEED=$1
             ;;
+        --traindiffs_test)
+	        shift
+            TEST=$1
+	        ;;
         *) 
             usage 
             exit 1