From 37e3e7775f78bbd42872932a61ccd0190dbfc01a Mon Sep 17 00:00:00 2001 From: Huihuo Zheng Date: Tue, 9 Jul 2024 16:12:56 -0500 Subject: [PATCH] Update custom_npz_reader.py --- .../plugins/experimental/src/reader/custom_npz_reader.py | 4 ++-- 1 file changed, 2 insertions(+), 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 index bdc4d790..9da296f5 100644 --- 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() \ No newline at end of file + return super().finalize()