diff --git a/context/cuvs-bench/run_benchmark.sh b/context/cuvs-bench/run_benchmark.sh index 17c79335..c93b21bb 100644 --- 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 index 50c5e3f2..5d9662cf 100644 --- 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")