n/a

PiperOrigin-RevId: 245754479
tensorflow · May 2, 2019 · 39d2b2a · 39d2b2a
1 parent d0a4c42
commit 39d2b2a
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/examples/half_plus_two/BUILD b/examples/half_plus_two/BUILD
@@ -17,6 +17,7 @@ licenses(["notice"])  # Apache 2.0 License
 py_binary(
     name = "export",
     srcs = ["export.py"],
+    python_version = "PY2",
     deps = [
         "//tensorflow_hub:expect_tensorflow_installed",
         "//tensorflow_hub",
@@ -27,6 +28,7 @@ py_test(
     name = "half_plus_two_test",
     srcs = ["half_plus_two_test.py"],
     data = [":export"],
+    python_version = "PY2",
     deps = [
         "//tensorflow_hub:expect_tensorflow_installed",
         "//tensorflow_hub",

diff --git a/examples/image_retraining/BUILD b/examples/image_retraining/BUILD
@@ -8,6 +8,7 @@ py_binary(
     srcs = [
         "retrain.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow_hub:expect_numpy_installed",

diff --git a/examples/text_embeddings/BUILD b/examples/text_embeddings/BUILD
@@ -17,6 +17,7 @@ licenses(["notice"])  # Apache 2.0
 py_binary(
     name = "export",
     srcs = ["export.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":export_lib"],
 )
@@ -36,6 +37,7 @@ py_library(
 py_test(
     name = "export_test",
     srcs = ["export_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":export_lib",

diff --git a/tensorflow_hub/BUILD b/tensorflow_hub/BUILD
@@ -77,6 +77,7 @@ py_library(
 py_test(
     name = "estimator_test",
     srcs = ["estimator_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["notsan"],  # b/111624588
     deps = [
@@ -100,6 +101,7 @@ py_library(
 py_test(
     name = "feature_column_test",
     srcs = ["feature_column_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":tensorflow_hub",
@@ -123,6 +125,7 @@ py_test(
     name = "compressed_module_resolver_test",
     size = "medium",
     srcs = ["compressed_module_resolver_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["nofixdeps"],
     deps = [
@@ -146,6 +149,7 @@ py_library(
 py_test(
     name = "image_util_test",
     srcs = ["image_util_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":image_util",
@@ -173,6 +177,7 @@ py_library(
 py_test(
     name = "module_test",
     srcs = ["module_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":module",
@@ -198,6 +203,7 @@ py_library(
 py_test(
     name = "saved_model_module_test",
     srcs = ["saved_model_module_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":saved_model_lib",
@@ -225,6 +231,7 @@ py_library(
 py_test(
     name = "native_module_test",
     srcs = ["native_module_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":all_protos_py_pb2",
@@ -250,6 +257,7 @@ py_test(
     name = "resolver_test",
     size = "medium",
     srcs = ["resolver_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":resolver",
@@ -274,6 +282,7 @@ py_test(
     data = [
         "testdata/half_plus_two_v1.tar.gz",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":tensorflow_hub",
@@ -299,6 +308,7 @@ py_test(
     name = "saved_model_lib_test",
     size = "small",
     srcs = ["saved_model_lib_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":saved_model_lib",
@@ -330,6 +340,7 @@ py_test(
     name = "tensor_info_test",
     size = "small",
     srcs = ["tensor_info_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":tensor_info",
@@ -355,6 +366,7 @@ py_library(
 py_test(
     name = "registry_test",
     srcs = ["registry_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":registry",
@@ -374,6 +386,7 @@ py_library(
 py_test(
     name = "keras_layer_test",
     srcs = ["keras_layer_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":tensorflow_hub",