Index: telemetry/telemetry/internal/platform/profiler/trace_profiler.py |
diff --git a/telemetry/telemetry/internal/platform/profiler/trace_profiler.py b/telemetry/telemetry/internal/platform/profiler/trace_profiler.py |
index d68ea7a0b7a3edab976a7e2850224aef2f09bb98..76b5632f5af0becc2ad589e47913ea6e2db4c9a6 100644 |
--- a/telemetry/telemetry/internal/platform/profiler/trace_profiler.py |
+++ b/telemetry/telemetry/internal/platform/profiler/trace_profiler.py |
@@ -7,8 +7,8 @@ import StringIO |
import zipfile |
from telemetry.internal.platform import profiler |
+from telemetry.timeline import chrome_trace_category_filter |
from telemetry.timeline import trace_data as trace_data_module |
-from telemetry.timeline import tracing_category_filter |
from telemetry.timeline import tracing_config |
@@ -25,8 +25,9 @@ class TraceProfiler(profiler.Profiler): |
categories_with_flow += ',%s' % categories |
config = tracing_config.TracingConfig() |
config.enable_chrome_trace = True |
- config.chrome_trace_config.SetTracingCategoryFilter( |
- tracing_category_filter.TracingCategoryFilter(categories_with_flow)) |
+ config.chrome_trace_config.SetCategoryFilter( |
+ chrome_trace_category_filter.ChromeTraceCategoryFilter( |
+ categories_with_flow)) |
self._browser_backend.StartTracing(config, timeout=10) |
@classmethod |