From 3da9761d256622088c962881a49a2ec4b8104a20 Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Fri, 23 Feb 2024 16:46:44 -0500 Subject: [PATCH] Use datasets as expected by docs (#1622) --- src/deepsparse/evaluation/cli.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/deepsparse/evaluation/cli.py b/src/deepsparse/evaluation/cli.py index 5f5f0fa9bb..9b10f5981c 100644 --- a/src/deepsparse/evaluation/cli.py +++ b/src/deepsparse/evaluation/cli.py @@ -23,7 +23,7 @@ MODEL_PATH A path to an ONNX model, local directory containing ONNX model (including all the auxiliary files) or a SparseZoo stub - -d DATASET, --dataset DATASET + -d DATASETS, --datasets DATASETS The dataset to evaluate on. The user may pass multiple datasets by passing the option multiple times. -i INTEGRATION, --integration INTEGRATION @@ -95,7 +95,7 @@ ) @click.option( "-d", - "--dataset", + "--datasets", type=str, multiple=True, help="The name of dataset to evaluate on. The user may pass multiple " @@ -163,7 +163,7 @@ @click.argument("integration_args", nargs=-1, type=click.UNPROCESSED) def main( model_path, - dataset, + datasets, integration, engine_type, save_path, @@ -182,7 +182,7 @@ def main( """ # join datasets to a list if multiple datasets are passed - datasets = list(dataset) if not isinstance(dataset, str) else dataset + datasets = list(datasets) if not isinstance(datasets, str) else datasets # format kwargs to a dict integration_args = parse_kwarg_tuples(integration_args)