| Index: tools/telemetry/telemetry/record_wpr.py
|
| diff --git a/tools/telemetry/telemetry/record_wpr.py b/tools/telemetry/telemetry/record_wpr.py
|
| index 6f1d2bd193bd61e81fb538a2f83a450d853de989..9a668f997626b1f3eb1a936015a90561a5ae29cc 100644
|
| --- a/tools/telemetry/telemetry/record_wpr.py
|
| +++ b/tools/telemetry/telemetry/record_wpr.py
|
| @@ -8,13 +8,13 @@
|
|
|
| from telemetry import benchmark
|
| from telemetry import story
|
| +from telemetry.core import discover
|
| from telemetry.core import util
|
| from telemetry.internal.browser import browser_options
|
| from telemetry.internal.results import results_options
|
| from telemetry.internal import story_runner
|
| from telemetry.internal.util import command_line
|
| from telemetry.page import page_test
|
| -from telemetry.util import classes_util
|
| from telemetry.util import wpr_modes
|
|
|
|
|
| @@ -71,13 +71,15 @@
|
| Returns:
|
| dict of {underscored_class_name: benchmark class}
|
| """
|
| - return classes_util.DiscoverClassesByClassName(base_dir, base_dir, cls)
|
| + return discover.DiscoverClasses(base_dir, base_dir, cls,
|
| + index_by_class_name=True)
|
|
|
|
|
| def _MaybeGetInstanceOfClass(target, base_dir, cls):
|
| if isinstance(target, cls):
|
| return target
|
| - return classes_util.MaybeGetInstanceOfClass(target, base_dir, base_dir, cls)
|
| + classes = _GetSubclasses(base_dir, cls)
|
| + return classes[target]() if target in classes else None
|
|
|
|
|
| def _PrintAllImpl(all_items, item_name, output_stream):
|
|
|