Skip to content

Commit

Permalink
fix misleading generator message (#231)
Browse files Browse the repository at this point in the history
The message should be `Generating HDF5 data` instead of NPZ
  • Loading branch information
rayandrew authored Oct 16, 2024
1 parent 3732663 commit 05d67d1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dlio_benchmark/data_generator/hdf5_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ def generate(self):
dim2 = dim[2*i+1]
records = np.random.randint(255, size=(dim1, dim2, self.num_samples), dtype=np.uint8)
out_path_spec = self.storage.get_uri(self._file_list[i])
progress(i+1, self.total_files_to_generate, "Generating NPZ Data")
progress(i+1, self.total_files_to_generate, "Generating HDF5 Data")
hf = h5py.File(out_path_spec, 'w')
hf.create_dataset('records', (self.num_samples, dim1, dim2), chunks=chunks, compression=compression,
compression_opts=compression_level, dtype=np.uint8, data=records)
Expand Down

0 comments on commit 05d67d1

Please sign in to comment.