Skip to content

Commit 68243e5

Browse files
authored
Use datasets as expected by docs (#1623)
1 parent 6267f19 commit 68243e5

File tree

1 file changed

+4
-4
lines changed
  • src/deepsparse/evaluation

1 file changed

+4
-4
lines changed

src/deepsparse/evaluation/cli.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
MODEL_PATH
2424
A path to an ONNX model, local directory containing ONNX model
2525
(including all the auxiliary files) or a SparseZoo stub
26-
-d DATASET, --dataset DATASET
26+
-d DATASETS, --datasets DATASETS
2727
The dataset to evaluate on. The user may pass multiple datasets
2828
by passing the option multiple times.
2929
-i INTEGRATION, --integration INTEGRATION
@@ -95,7 +95,7 @@
9595
)
9696
@click.option(
9797
"-d",
98-
"--dataset",
98+
"--datasets",
9999
type=str,
100100
multiple=True,
101101
help="The name of dataset to evaluate on. The user may pass multiple "
@@ -163,7 +163,7 @@
163163
@click.argument("integration_args", nargs=-1, type=click.UNPROCESSED)
164164
def main(
165165
model_path,
166-
dataset,
166+
datasets,
167167
integration,
168168
engine_type,
169169
save_path,
@@ -182,7 +182,7 @@ def main(
182182
183183
"""
184184
# join datasets to a list if multiple datasets are passed
185-
datasets = list(dataset) if not isinstance(dataset, str) else dataset
185+
datasets = list(datasets) if not isinstance(datasets, str) else datasets
186186
# format kwargs to a dict
187187
integration_args = parse_kwarg_tuples(integration_args)
188188

0 commit comments

Comments
 (0)