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 ccb12f67bfc8f3b5ce313a64fcfc06de5c97592f..73cd0288a1b9e9ced9032f4aaf6cd91802d6adbd 100644 |
--- a/content/test/gpu/page_sets/__init__.py |
+++ b/content/test/gpu/page_sets/__init__.py |
@@ -7,11 +7,7 @@ import sys |
from telemetry.core import discover |
from telemetry.story import story_set |
- |
-# 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 discover.DiscoverClasses( |
- start_dir, top_level_dir, base_class).values(): |
- setattr(sys.modules[__name__], cls.__name__, cls) |
+from page_sets.gpu_process_tests import GpuProcessTestsStorySet |
+from page_sets.gpu_rasterization_tests import GpuRasterizationTestsStorySet |
+from page_sets.memory_tests import MemoryTestsStorySet |
+from page_sets.pixel_tests import PixelTestsStorySet |