@@ -61,7 +61,9 @@ serving_proto_library_py(
6161 name = "input_proto_py_pb2" ,
6262 srcs = ["input.proto" ],
6363 proto_library = "input_proto" ,
64- deps = [],
64+ deps = [
65+ "@org_tensorflow//tensorflow/core:protos_all_py" ,
66+ ],
6567)
6668
6769serving_proto_library (
@@ -124,8 +126,10 @@ py_library(
124126 name = "prediction_service_proto_py_pb2" ,
125127 srcs = ["prediction_service_pb2.py" ],
126128 deps = [
129+ ":classification_proto_py_pb2" ,
127130 ":get_model_metadata_proto_py_pb2" ,
128131 ":predict_proto_py_pb2" ,
132+ ":regression_proto_py_pb2" ,
129133 ],
130134)
131135
@@ -145,7 +149,11 @@ serving_proto_library_py(
145149 name = "classification_proto_py_pb2" ,
146150 srcs = ["classification.proto" ],
147151 proto_library = "classification_proto" ,
148- deps = [],
152+ deps = [
153+ ":input_proto_py_pb2" ,
154+ ":model_proto_py_pb2" ,
155+ "@org_tensorflow//tensorflow/core:protos_all_py" ,
156+ ],
149157)
150158
151159serving_proto_library (
@@ -164,7 +172,11 @@ serving_proto_library_py(
164172 name = "regression_proto_py_pb2" ,
165173 srcs = ["regression.proto" ],
166174 proto_library = "regression_proto" ,
167- deps = [],
175+ deps = [
176+ ":input_proto_py_pb2" ,
177+ ":model_proto_py_pb2" ,
178+ "@org_tensorflow//tensorflow/core:protos_all_py" ,
179+ ],
168180)
169181
170182cc_library (
0 commit comments