Index: tools/perf/benchmarks/benchmark_smoke_unittest.py |
diff --git a/tools/perf/benchmarks/benchmark_smoke_unittest.py b/tools/perf/benchmarks/benchmark_smoke_unittest.py |
index 69ffa268fbfaf81ae0dca5977e4f43abdd593423..b8ccb0cbb12270174675277dbd1d2a926fb74d46 100644 |
--- a/tools/perf/benchmarks/benchmark_smoke_unittest.py |
+++ b/tools/perf/benchmarks/benchmark_smoke_unittest.py |
@@ -104,11 +104,11 @@ def load_tests(loader, standard_tests, pattern): |
benchmarks_dir = os.path.dirname(__file__) |
top_level_dir = os.path.dirname(benchmarks_dir) |
- # Using the default of |index_by_class_name=False| means that if a module |
- # has multiple benchmarks, only the last one is returned. |
+ # Using the default of |one_class_per_module=True| means that if a module |
+ # has multiple benchmarks, only the first one is returned. |
all_benchmarks = discover.DiscoverClasses( |
benchmarks_dir, top_level_dir, benchmark_module.Benchmark, |
- index_by_class_name=False).values() |
+ one_class_per_module=True) |
for benchmark in all_benchmarks: |
if sys.modules[benchmark.__module__] in _BLACK_LIST_TEST_MODULES: |
continue |