Refactoring: Split logic from generate_examples py_binary target.

PiperOrigin-RevId: 239526811
tensorflow · Mar 21, 2019 · 17ca654 · 17ca654
1 parent dca741b
commit 17ca654
Show file tree

Hide file tree

Showing 3 changed files with 4,605 additions and 4,501 deletions.
diff --git a/tensorflow/lite/testing/BUILD b/tensorflow/lite/testing/BUILD
@@ -70,9 +70,9 @@ test_suite(
     ],
 )
 
-py_binary(
-    name = "generate_examples",
-    srcs = ["generate_examples.py"],
+py_library(
+    name = "generate_examples_lib",
+    srcs = ["generate_examples_lib.py"],
     data = [
         "//tensorflow/lite/toco",
     ],
@@ -87,6 +87,21 @@ py_binary(
     ],
 )
 
+py_binary(
+    name = "generate_examples",
+    srcs = ["generate_examples.py"],
+    data = [
+        "//tensorflow/lite/toco",
+    ],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":generate_examples_lib",
+        "//tensorflow:tensorflow_py",
+        "//third_party/py/numpy",
+        "@six_archive//:six",
+    ],
+)
+
 py_library(
     name = "generate_examples_report",
     srcs = ["generate_examples_report.py"],