FIX use new data-export flag

rapidsai · Oct 16, 2024 · a3bb682 · a3bb682
1 parent 6b61b5d
commit a3bb682
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/context/cuvs-bench/run_benchmark.sh b/context/cuvs-bench/run_benchmark.sh
@@ -32,8 +32,8 @@ fi
 # (2) build and search index
 python -m cuvs_bench.run  ${DATASET_ARG} --dataset-path /data/benchmarks/datasets --force ${RUN_ARGS}
 
-# (3) export data
-python -m cuvs_bench.data_export  ${DATASET_ARG} --dataset-path /data/benchmarks/datasets
+# (3) export data again in case a benchmark crashed
+python -m cuvs_bench.run  ${DATASET_ARG} --dataset-path /data/benchmarks/datasets --data-export
 
 # Extract the algorithms from the run command to use in the plot command
 ALGOS=$(grep -oP "algorithms\s+\K(\w+,?\w+)" <<< "$RUN_ARGS")

diff --git a/context/cuvs-bench/run_benchmarks_preloaded_datasets.sh b/context/cuvs-bench/run_benchmarks_preloaded_datasets.sh
@@ -40,7 +40,7 @@ fi
 python -m cuvs_bench.run  ${DATASET_ARG} --dataset-path $DATASET_PATH ${RUN_ARGS}
 
 # (3) export data
-python -m cuvs_bench.data_export  ${DATASET_ARG} --dataset-path $DATASET_PATH
+python -m cuvs_bench.run  ${DATASET_ARG} --dataset-path $DATASET_PATH --data-export
 
 # Extract the algorithms from the run command to use in the plot command
 ALGOS=$(grep -oP "algorithms\s+\K(\w+,?\w+)" <<< "$RUN_ARGS")