Skip to content

Commit

Permalink
[data][tests-only] Fix nightly microbenchmark for read_images #39609
Browse files Browse the repository at this point in the history
Fix a test error in read_images_comparison_microbenchmark_single_node.

Signed-off-by: Stephanie Wang <swang@cs.berkeley.edu>
  • Loading branch information
stephanie-wang committed Sep 13, 2023
1 parent ed32450 commit 3e49f5d
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -530,6 +530,7 @@ def __getitem__(self, idx: int) -> Any:
ray_dataset = ray.data.read_datasource(
mds_source, paths=args.mosaic_data_root
)
ray_dataset = ray_dataset.map(crop_and_flip_image)
for i in range(args.num_epochs):
iterate(
ray_dataset.iter_torch_batches(batch_size=args.batch_size),
Expand Down

0 comments on commit 3e49f5d

Please sign in to comment.