Index: tools/telemetry/telemetry/benchmark.py |
diff --git a/tools/telemetry/telemetry/benchmark.py b/tools/telemetry/telemetry/benchmark.py |
index 6ea2196659b72e58275039311f03307a69156d51..0f2616f5301df468ea1eadea454954a579ea4c63 100644 |
--- a/tools/telemetry/telemetry/benchmark.py |
+++ b/tools/telemetry/telemetry/benchmark.py |
@@ -201,7 +201,7 @@ class Benchmark(command_line.Command): |
Otherwise, a TimelineBasedMeasurement instance. |
""" |
is_page_test = issubclass(self.test, page_test.PageTest) |
- is_tbm = self.test == timeline_based_measurement.TimelineBasedMeasurement |
+ is_tbm = True |
if not is_page_test and not is_tbm: |
raise TypeError('"%s" is not a PageTest or a TimelineBasedMeasurement.' % |
self.test.__name__) |
@@ -213,7 +213,7 @@ class Benchmark(command_line.Command): |
opts = self.CreateTimelineBasedMeasurementOptions() |
self.SetupTraceRerunOptions(options, opts) |
- return timeline_based_measurement.TimelineBasedMeasurement(opts) |
+ return self.test(opts) |
def CreateStorySet(self, options): |
"""Creates the instance of StorySet used to run the benchmark. |