Index: tools/perf/core/perf_data_generator.py |
diff --git a/tools/perf/core/perf_data_generator.py b/tools/perf/core/perf_data_generator.py |
index 3feac7eec9afc8337c42e16449090b0de80e18e5..0bf9e5de5465ec481d6647e643973415bdcf8a77 100755 |
--- a/tools/perf/core/perf_data_generator.py |
+++ b/tools/perf/core/perf_data_generator.py |
@@ -23,9 +23,10 @@ from core import path_util |
path_util.AddTelemetryToPath() |
from telemetry import benchmark as benchmark_module |
-from telemetry.core import discover |
from telemetry import decorators |
+from py_utils import discover |
+ |
from core.sharding_map_generator import load_benchmark_sharding_map |