Skip to content
This repository has been archived by the owner on Jan 10, 2025. It is now read-only.

Commit

Permalink
qa
Browse files Browse the repository at this point in the history
  • Loading branch information
floriankrb committed Nov 30, 2023
1 parent 014c394 commit 868e1ba
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions scripts/speed-test.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,14 +12,17 @@ def main():
parser = argparse.ArgumentParser()
parser.add_argument(
"path",
help="Path to the dataset, use s3:// for S3 storage, http:// for HTTP storage, and /path/to/dataset for local storage",
help=(
"Path to the dataset, use s3:// for S3 storage, "
"http:// for HTTP storage, and /path/to/dataset for local storage"
),
)
parser.add_argument(
"--shuffle",
action="store_true",
help="Whether to shuffle the dataset",
)

args = parser.parse_args()

ds = open_dataset(args.path)
Expand Down

0 comments on commit 868e1ba

Please sign in to comment.