fix missing import for chunking. (#223)

argonne-lcf · Aug 28, 2024 · d6924f3 · d6924f3
1 parent ec22aec
commit d6924f3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dlio_benchmark/data_generator/hdf5_generator.py b/dlio_benchmark/data_generator/hdf5_generator.py
@@ -17,6 +17,7 @@
 
 import h5py
 import numpy as np
+import math # is needed for chunking case.
 
 from dlio_benchmark.common.enumerations import Compression
 from dlio_benchmark.data_generator.data_generator import DataGenerator