Update custom_npz_reader.py

argonne-lcf · Jul 9, 2024 · 37e3e77 · 37e3e77
1 parent 1b1bac6
commit 37e3e77
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dlio_benchmark/plugins/experimental/src/reader/custom_npz_reader.py b/dlio_benchmark/plugins/experimental/src/reader/custom_npz_reader.py
@@ -19,7 +19,7 @@
 from dlio_benchmark.common.constants import MODULE_DATA_READER
 from dlio_benchmark.reader.reader_handler import FormatReader
 
-from dftracer.logger import fn_interceptor as Profile
+from dlio_benchmark.utils.utility import Profile
 
 dlp = Profile(MODULE_DATA_READER)
 
@@ -58,4 +58,4 @@ def read_index(self, image_idx, step):
 
     @dlp.log
     def finalize(self):
-        return super().finalize()
+        return super().finalize()