Update custom_torch_data_loader.py [ci-skip]

argonne-lcf · Jul 9, 2024 · 8d4f473 · 8d4f473
1 parent 37e3e77
commit 8d4f473
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlio_benchmark/plugins/experimental/src/data_loader/custom_torch_data_loader.py b/dlio_benchmark/plugins/experimental/src/data_loader/custom_torch_data_loader.py
@@ -9,7 +9,7 @@
 from dlio_benchmark.data_loader.base_data_loader import BaseDataLoader
 from dlio_benchmark.reader.reader_factory import ReaderFactory
 from dlio_benchmark.utils.utility import utcnow, DLIOMPI
-from dftracer.logger import fn_interceptor as Profile
+from dlio_benchmark.utils.utility import Profile
 
 dlp = Profile(MODULE_DATA_LOADER)