Index: tools/telemetry/telemetry/benchmark_runner.py |
diff --git a/tools/telemetry/telemetry/benchmark_runner.py b/tools/telemetry/telemetry/benchmark_runner.py |
index 44518d1e291252e221c074c4e5a3e3b17298a7cb..3ad52da462a85224c0f59406dccc20b774a9f405 100644 |
--- a/tools/telemetry/telemetry/benchmark_runner.py |
+++ b/tools/telemetry/telemetry/benchmark_runner.py |
@@ -15,11 +15,11 @@ import os |
import sys |
from telemetry import benchmark |
-from telemetry.core import discover |
from telemetry import decorators |
from telemetry.internal.browser import browser_finder |
from telemetry.internal.browser import browser_options |
from telemetry.internal.util import command_line |
+from telemetry.util import classes_util |
def PrintBenchmarkList(benchmarks, possible_browser, output_pipe=sys.stdout): |
@@ -291,10 +291,9 @@ def _MatchingCommands(string): |
def _Benchmarks(environment): |
benchmarks = [] |
for search_dir in environment.benchmark_dirs: |
- benchmarks += discover.DiscoverClasses(search_dir, |
+ benchmarks += classes_util.DiscoverClasses(search_dir, |
environment.top_level_dir, |
- benchmark.Benchmark, |
- index_by_class_name=True).values() |
+ benchmark.Benchmark) |
return benchmarks |
def _MatchBenchmarkName(input_benchmark_name, environment, exact_matches=True): |