From ec3c9d2103db2e430f83903112e0f23dd3611d23 Mon Sep 17 00:00:00 2001
From: "candy.dc" <candy.dc@alibaba-inc.com>
Date: Mon, 15 Jan 2024 19:15:27 +0800
Subject: [PATCH] [Serving] Fix syntax error in generate timeline tool.

Signed-off-by: candy.dc <candy.dc@alibaba-inc.com>
---
 serving/tools/timeline/gen_timeline.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/serving/tools/timeline/gen_timeline.py b/serving/tools/timeline/gen_timeline.py
index f055e473fa0..d56c1b39897 100644
--- a/serving/tools/timeline/gen_timeline.py
+++ b/serving/tools/timeline/gen_timeline.py
@@ -1,6 +1,6 @@
 import sys
-import config_pb2
-import timeline
+from tensorflow.core.protobuf import config_pb2
+from tensorflow.python.client import timeline
 
 def gen_timeline(src_name, dest_name):
   run_metadata = config_pb2.RunMetadata()