Fix properly apply save-dir option

sony · Aug 28, 2023 · 4adaa11 · 4adaa11
1 parent ce80aad
commit 4adaa11
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/bin/evaluate_algorithm b/bin/evaluate_algorithm
@@ -156,7 +156,11 @@ do
 done
 
 REPRODUCTION_CODE_DIR="${ROOT_DIR}/reproductions/algorithms/${BASE_ENV_NAME}/${ALGO_NAME}"
-SAVE_DIR=$REPRODUCTION_CODE_DIR
+if [ -n "$SAVE_DIR" ]; then
+    SAVE_DIR=$REPRODUCTION_CODE_DIR/$SAVE_DIR
+else
+    SAVE_DIR=$REPRODUCTION_CODE_DIR
+fi
 
 for ENV_NAME in ${ENVS}
 do