Skip to content

Commit 342eaef

Browse files
Fix an include and BUILD files to be OSS compatible.
Change: 146931707
1 parent 21d5a91 commit 342eaef

File tree

1 file changed

+15
-3
lines changed
  • tensorflow_serving/apis

1 file changed

+15
-3
lines changed

tensorflow_serving/apis/BUILD

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

6769
serving_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

151159
serving_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

170182
cc_library(

0 commit comments

Comments
 (0)