| Index: tools/chrome_proxy/live_tests/pagesets/__init__.py
|
| diff --git a/tools/chrome_proxy/live_tests/pagesets/__init__.py b/tools/chrome_proxy/live_tests/pagesets/__init__.py
|
| index 45ce4af9df80c564b044b526df9669c00e3a809c..a67d12d383c435d0cbf7b80df98e1b6faef0eb4d 100644
|
| --- a/tools/chrome_proxy/live_tests/pagesets/__init__.py
|
| +++ b/tools/chrome_proxy/live_tests/pagesets/__init__.py
|
| @@ -6,15 +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' StorySet classes.
|
| start_dir = os.path.dirname(os.path.abspath(__file__))
|
| top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir))
|
| 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)
|
|
|
|
|