fix

mlcommons · Apr 10, 2024 · 59aea77 · 59aea77
1 parent 4721be4
commit 59aea77
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/docker/scripts/startup.sh b/docker/scripts/startup.sh
@@ -137,6 +137,8 @@ while [ "$1" != "" ]; do
             ADDITIONAL_REQUIREMENTS_PATH=$1
             ;;
         *) 
+            shift
+            echo "Unrecognized flag $1"
             usage 
             exit 1
             ;;
@@ -147,6 +149,7 @@ done
 # Optionnally install addtional dependencies
 if [[ -n ${ADDITIONAL_REQUIREMENTS_PATH} ]]; then
     pip install -r ${ADDITIONAL_REQUIREMENTS_PATH}
+fi
 
 # Run traindiff tests
 if [[ ${TEST} == "true" ]]; then
@@ -324,4 +327,4 @@ then
 fi
 
 echo "Exiting with $RETURN_CODE"
-exit $RETURN_CODE
+exit $RETURN_CODE