Index: tools/perf/page_sets/__init__.py |
diff --git a/tools/perf/page_sets/__init__.py b/tools/perf/page_sets/__init__.py |
index 65daf2f00485efec7d5d1c81a30c17b8c9b5975b..d585f3271d79f6dee010f18099573ed618a7e5a7 100644 |
--- a/tools/perf/page_sets/__init__.py |
+++ b/tools/perf/page_sets/__init__.py |
@@ -6,14 +6,14 @@ import inspect |
import os |
import sys |
-from telemetry.core import discover |
from telemetry import story |
+from telemetry.util import classes_util |
# Import all submodules' PageSet classes. |
start_dir = os.path.dirname(os.path.abspath(__file__)) |
top_level_dir = os.path.dirname(start_dir) |
base_class = story.StorySet |
-for cls in discover.DiscoverClasses( |
- start_dir, top_level_dir, base_class).values(): |
+for cls in classes_util.DiscoverClasses( |
+ start_dir, top_level_dir, base_class): |
setattr(sys.modules[__name__], cls.__name__, cls) |