Index: content/test/gpu/page_sets/__init__.py |
diff --git a/content/test/gpu/page_sets/__init__.py b/content/test/gpu/page_sets/__init__.py |
index 7d6dd796b8ec9dd39bf39ef686fedf746e7a0def..ccb12f67bfc8f3b5ce313a64fcfc06de5c97592f 100644 |
--- a/content/test/gpu/page_sets/__init__.py |
+++ b/content/test/gpu/page_sets/__init__.py |
@@ -4,13 +4,14 @@ |
import os |
import sys |
+from telemetry.core import discover |
from telemetry.story import story_set |
-from telemetry.util import classes_util |
# Import all submodules' StorySet classes. |
start_dir = os.path.dirname(os.path.abspath(__file__)) |
top_level_dir = os.path.dirname(start_dir) |
base_class = story_set.StorySet |
-for cls in classes_util.DiscoverClasses(start_dir, top_level_dir, base_class): |
+for cls in discover.DiscoverClasses( |
+ start_dir, top_level_dir, base_class).values(): |
setattr(sys.modules[__name__], cls.__name__, cls) |