Drop Python 2 tests.

PiperOrigin-RevId: 317342218

Author: csferng

neural_structured_learning/estimator/BUILD

@@ -15,8 +15,7 @@
 # Description:
 # Build rules for Estimator APIs in Neural Structured Learning.
 
-# Placeholder for internal Python strict compatibility macro.
-load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
+# Placeholder for internal Python strict & test compatibility macro.
 
 package(
     default_visibility = ["//visibility:public"],
@@ -46,11 +45,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "adversarial_regularization_test",
     srcs = ["adversarial_regularization_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":estimator",
         # package absl/testing:parameterized
@@ -73,11 +71,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "graph_regularization_test",
     srcs = ["graph_regularization_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":estimator",
         # package protobuf,

neural_structured_learning/keras/BUILD

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Placeholder for internal Python strict compatibility macro.
-load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
+# Placeholder for internal Python strict & test compatibility macro.
 
 # Description:
 # Build rules for Keras APIs in Neural Structured Learning.
@@ -48,11 +47,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "adversarial_regularization_multi_device_test",
     srcs = ["adversarial_regularization_multi_device_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":adversarial_regularization",
         "//neural_structured_learning/configs",
@@ -61,11 +59,10 @@ py2and3_strict_test(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "adversarial_regularization_test",
     srcs = ["adversarial_regularization_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":adversarial_regularization",
         # package absl/testing:parameterized
@@ -88,12 +85,11 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "graph_regularization_test",
     timeout = "long",
     srcs = ["graph_regularization_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":graph_regularization",
         # package protobuf,

neural_structured_learning/keras/layers/BUILD

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Placeholder for internal Python strict compatibility macro.
-load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
+# Placeholder for internal Python strict & test compatibility macro.
 
 # Description:
 # Build rules for custom Keras layers in Neural Structured Learning.
@@ -35,13 +34,12 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "layers_test",
     size = "medium",
     srcs = [
         "layers_test.py",
     ],
-    python_version = "PY3",
     deps = [
         ":layers",
         # package absl/testing:parameterized
@@ -64,13 +62,12 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "neighbor_features_test",
     srcs = [
         "neighbor_features_test.py",
     ],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":neighbor_features",
         # package absl/testing:parameterized
@@ -95,11 +92,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "pairwise_distance_test",
     srcs = ["pairwise_distance_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":pairwise_distance",
         "//neural_structured_learning/configs",

neural_structured_learning/lib/BUILD

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Placeholder for internal Python strict compatibility macro.
-load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
+# Placeholder for internal Python strict & test compatibility macro.
 
 # Description:
 # Build rules for libraries and ops in Neural Structured Learning.
@@ -60,11 +59,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "adversarial_neighbor_test",
     srcs = ["adversarial_neighbor_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":adversarial_neighbor",
         # package absl/testing:parameterized
@@ -85,11 +83,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "distances_test",
     srcs = ["distances_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":distances",
         "//neural_structured_learning/configs",
@@ -109,11 +106,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "regularizer_test",
     srcs = ["regularizer_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":regularizer",
         "//neural_structured_learning/configs",
@@ -133,12 +129,11 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "utils_test",
     size = "small",
     srcs = ["utils_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":utils",
         # package absl/testing:parameterized

neural_structured_learning/tools/BUILD

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Placeholder for internal Python strict compatibility macro.
-load("//devtools/python/blaze:python3.bzl", "py2and3_strict_test")
+# Placeholder for internal Python strict & test compatibility macro.
 
 # Description:
 # Build rules for tools in Neural Structured Learning.
@@ -46,11 +45,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "graph_utils_test",
     srcs = ["graph_utils_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":graph_utils",
         # package absl/testing:absltest
@@ -71,11 +69,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "build_graph_test",
     srcs = ["build_graph_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":build_graph_lib",
         ":graph_utils",
@@ -108,11 +105,10 @@ py_library(
     ],
 )
 
-py2and3_strict_test(
+py_test(
     name = "pack_nbrs_test",
     srcs = ["pack_nbrs_test.py"],
-    python_version = "PY3",
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":graph_utils",
         ":pack_nbrs_lib",