tp: Create Runner classes
Bug:255535171
Change-Id: I202775792f79b2ef2d1be16b77e8ee0c066e1025
diff --git a/tools/serialize_test_trace.py b/tools/serialize_test_trace.py
index 49cdc68..6c7fa85 100755
--- a/tools/serialize_test_trace.py
+++ b/tools/serialize_test_trace.py
@@ -21,7 +21,10 @@
import os
import sys
-from proto_utils import serialize_python_trace, serialize_textproto_trace
+ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+sys.path.append(ROOT_DIR)
+
+from python.generators.diff_tests.utils import serialize_textproto_trace, serialize_python_trace
def main():
@@ -55,7 +58,8 @@
trace_path = args.trace_path
if trace_path.endswith('.py'):
- serialize_python_trace(trace_descriptor_path, trace_path, sys.stdout.buffer)
+ serialize_python_trace(ROOT_DIR, trace_descriptor_path, trace_path,
+ sys.stdout.buffer)
elif trace_path.endswith('.textproto'):
serialize_textproto_trace(trace_descriptor_path, extension_descriptors,
trace_path, sys.stdout.buffer)
