internal change

PiperOrigin-RevId: 316553522

Author: csferng

neural_structured_learning/estimator/BUILD

@@ -16,7 +16,7 @@
 # Build rules for Estimator APIs in Neural Structured Learning.
 
 # Placeholder for internal Python strict compatibility macro.
-# Placeholder for internal Python version compatibility macro.
+load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
 
 package(
     default_visibility = ["//visibility:public"],
@@ -46,9 +46,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "adversarial_regularization_test",
     srcs = ["adversarial_regularization_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":estimator",
@@ -72,9 +73,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "graph_regularization_test",
     srcs = ["graph_regularization_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":estimator",

neural_structured_learning/keras/BUILD

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # Placeholder for internal Python strict compatibility macro.
-# Placeholder for internal Python version compatibility macro.
+load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
 
 # Description:
 # Build rules for Keras APIs in Neural Structured Learning.
@@ -48,9 +48,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "adversarial_regularization_multi_device_test",
     srcs = ["adversarial_regularization_multi_device_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":adversarial_regularization",
@@ -60,9 +61,10 @@ py_test(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "adversarial_regularization_test",
     srcs = ["adversarial_regularization_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":adversarial_regularization",
@@ -86,10 +88,11 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "graph_regularization_test",
     timeout = "long",
     srcs = ["graph_regularization_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":graph_regularization",

neural_structured_learning/keras/layers/BUILD

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # Placeholder for internal Python strict compatibility macro.
-# Placeholder for internal Python version compatibility macro.
+load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
 
 # Description:
 # Build rules for custom Keras layers in Neural Structured Learning.
@@ -35,12 +35,13 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "layers_test",
     size = "medium",
     srcs = [
         "layers_test.py",
     ],
+    python_version = "PY3",
     deps = [
         ":layers",
         # package absl/testing:parameterized
@@ -63,11 +64,12 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "neighbor_features_test",
     srcs = [
         "neighbor_features_test.py",
     ],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":neighbor_features",
@@ -93,9 +95,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "pairwise_distance_test",
     srcs = ["pairwise_distance_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":pairwise_distance",

neural_structured_learning/lib/BUILD

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # Placeholder for internal Python strict compatibility macro.
-# Placeholder for internal Python version compatibility macro.
+load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
 
 # Description:
 # Build rules for libraries and ops in Neural Structured Learning.
@@ -60,9 +60,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "adversarial_neighbor_test",
     srcs = ["adversarial_neighbor_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":adversarial_neighbor",
@@ -84,9 +85,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "distances_test",
     srcs = ["distances_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":distances",
@@ -107,9 +109,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "regularizer_test",
     srcs = ["regularizer_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":regularizer",
@@ -130,10 +133,11 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "utils_test",
     size = "small",
     srcs = ["utils_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":utils",

neural_structured_learning/tools/BUILD

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # Placeholder for internal Python strict compatibility macro.
-# Placeholder for internal Python version compatibility macro.
+load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
 
 # Description:
 # Build rules for tools in Neural Structured Learning.
@@ -46,9 +46,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "graph_utils_test",
     srcs = ["graph_utils_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":graph_utils",
@@ -70,9 +71,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "build_graph_test",
     srcs = ["build_graph_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":build_graph_lib",
@@ -106,9 +108,10 @@ py_library(
     ],
 )
 
-py_test(
+py2and3_strict_test(
     name = "pack_nbrs_test",
     srcs = ["pack_nbrs_test.py"],
+    python_version = "PY3",
     srcs_version = "PY2AND3",
     deps = [
         ":graph_utils",