Index: telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py |
diff --git a/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py b/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py |
index 3e06bfe31ae7141c10d9f5fcd6c73467371d751c..1c6dc065f95f5fa39b3deff074b1e8f355747d99 100644 |
--- a/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py |
+++ b/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py |
@@ -10,6 +10,8 @@ from devil.android import battery_utils |
from telemetry.internal.platform.tracing_agent import battor_tracing_agent |
from telemetry.timeline import trace_data |
from telemetry.timeline import tracing_config |
+from tracing.trace_data import trace_data |
+ |
_BATTOR_RETURN = 'fake\nbattor\ndata' |