Adding python targets for config protos

PiperOrigin-RevId: 378649319
tensorflow · Jun 10, 2021 · 757c3a6 · 757c3a6
1 parent 56ae5a6
commit 757c3a6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/tensorflow_serving/config/BUILD b/tensorflow_serving/config/BUILD
@@ -63,6 +63,11 @@ serving_proto_library(
     ],
 )
 
+serving_proto_library_py(
+    name = "platform_config_py_pb2",
+    proto_library = "platform_config_proto",
+)
+
 serving_proto_library(
     name = "log_collector_config_proto",
     srcs = ["log_collector_config.proto"],

diff --git a/tensorflow_serving/servables/tensorflow/BUILD b/tensorflow_serving/servables/tensorflow/BUILD
@@ -1,7 +1,7 @@
 # Description: TensorFlow servables.
 
 load("//tensorflow_serving:oss_or_google.bzl", "if_google", "if_oss")
-load("//tensorflow_serving:serving.bzl", "serving_proto_library", "serving_tensorflow_proto_dep")
+load("//tensorflow_serving:serving.bzl", "serving_proto_library", "serving_proto_library_py", "serving_tensorflow_proto_dep")
 
 package(
     default_visibility = [
@@ -38,6 +38,12 @@ serving_proto_library(
     ],
 )
 
+serving_proto_library_py(
+    name = "session_bundle_config_py_pb2",
+    proto_library = "session_bundle_config_proto",
+    visibility = ["//visibility:public"],
+)
+
 cc_library(
     name = "resource_estimator",
     hdrs = ["resource_estimator.h"],
@@ -252,6 +258,12 @@ serving_proto_library(
     ],
 )
 
+serving_proto_library_py(
+    name = "saved_model_bundle_source_adapter_py_pb2",
+    proto_library = "saved_model_bundle_source_adapter_proto",
+    visibility = ["//visibility:public"],
+)
+
 cc_library(
     name = "simple_servers",
     srcs = ["simple_servers.cc"],