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 17bcdfeb6a6a16d1f14adc91d1bafac1d7ea1e1c..255190f2ebc9bad04f1d3ea1e2d1fe3add9be5a8 100644 |
--- a/tools/perf/benchmarks/benchmark_smoke_unittest.py |
+++ b/tools/perf/benchmarks/benchmark_smoke_unittest.py |
@@ -102,11 +102,11 @@ |
benchmarks_dir = os.path.dirname(__file__) |
top_level_dir = os.path.dirname(benchmarks_dir) |
- # Using the default of |one_class_per_module=True| means that if a module |
- # has multiple benchmarks, only the first one is returned. |
+ # Using the default of |index_by_class_name=False| means that if a module |
+ # has multiple benchmarks, only the last one is returned. |
all_benchmarks = discover.DiscoverClasses( |
benchmarks_dir, top_level_dir, benchmark_module.Benchmark, |
- one_class_per_module=True) |
+ index_by_class_name=False).values() |
for benchmark in all_benchmarks: |
if sys.modules[benchmark.__module__] in _BLACK_LIST_TEST_MODULES: |
continue |