Index: tools/perf/page_sets/__init__.py |
diff --git a/tools/perf/page_sets/__init__.py b/tools/perf/page_sets/__init__.py |
index 5a112350469c309e250c10a3f6a3a9aac4cad9b3..7504b80dcbf109193bae7ed8435e17a907ae1705 100644 |
--- a/tools/perf/page_sets/__init__.py |
+++ b/tools/perf/page_sets/__init__.py |
@@ -13,8 +13,9 @@ from telemetry import story |
# 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 |
+base_classes = [story.StorySet, story.expectations.StoryExpectations] |
-for cls in discover.DiscoverClasses( |
- start_dir, top_level_dir, base_class).values(): |
- setattr(sys.modules[__name__], cls.__name__, cls) |
+for base_class in base_classes: |
+ for cls in discover.DiscoverClasses( |
+ start_dir, top_level_dir, base_class).values(): |
+ setattr(sys.modules[__name__], cls.__name__, cls) |