Fix deps error for evaluation_stages python library

PiperOrigin-RevId: 307934339 Change-Id: I1a59527899ed0de39dbd81c6f151a75c418be843
2020-04-22 17:02:37 -07:00 · 2020-04-22 17:02:37 -07:00 · d76bbee134
commit d76bbee134
parent 7bde3d9d14
1 changed files with 9 additions and 0 deletions
--- a/tensorflow/lite/tools/evaluation/proto/BUILD
+++ b/tensorflow/lite/tools/evaluation/proto/BUILD
@ -47,6 +47,15 @@ tf_proto_library_py(
    srcs = [
        "evaluation_stages.proto",
    ],
+    protodeps = [":preprocessing_steps"],
+    visibility = ["//visibility:public"],
+)
+
+tf_proto_library_py(
+    name = "preprocessing_steps",
+    srcs = [
+        "preprocessing_steps.proto",
+    ],
    visibility = ["//visibility:public"],
 )