Update run_predict_ensemble.sh

Co-authored-by: Bryn Noel Ubald <55503826+bnubald@users.noreply.github.com>
icenet-ai · Aug 20, 2024 · fcd47ab · fcd47ab
1 parent 5f2289b
commit fcd47ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_predict_ensemble.sh b/run_predict_ensemble.sh
@@ -16,7 +16,7 @@ ENSEMBLE_ARGS=""
 TRAIN_IDENT=""
 ENSEMBLE_SEEDS_DEFAULT=42,46,45
 
-while getopts ":b:df:i:lm:p:r:x" opt; do
+while getopts ":b:c:df:i:lm:p:r:x" opt; do
   case "$opt" in
     b)  ENSEMBLE_ARGS="${ENSEMBLE_ARGS}arg_batch=$OPTARG ";;
     c)  ENSEMBLE_ARGS="${ENSEMBLE_ARGS}cluster=$OPTARG ";;