| 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)
|
|
|