Skip to content

Commit

Permalink
Use datasets as expected by docs (#1623)
Browse files Browse the repository at this point in the history
  • Loading branch information
rahul-tuli committed Feb 23, 2024
1 parent 6267f19 commit 68243e5
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/deepsparse/evaluation/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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 "
Expand Down Expand Up @@ -163,7 +163,7 @@
@click.argument("integration_args", nargs=-1, type=click.UNPROCESSED)
def main(
model_path,
dataset,
datasets,
integration,
engine_type,
save_path,
Expand All @@ -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)

Expand Down

0 comments on commit 68243e5

Please sign in to comment.