Declare dropped Python 2 support in BUILD files.

PiperOrigin-RevId: 324640050

Author: alanchiao

tensorflow_model_optimization/BUILD

@@ -15,7 +15,7 @@ exports_files(["LICENSE"])
 py_library(
     name = "tensorflow_model_optimization",
     srcs = ["__init__.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         "//tensorflow_model_optimization/python",
         "//tensorflow_model_optimization/python/core/api",

tensorflow_model_optimization/python/BUILD

@@ -7,7 +7,7 @@ exports_files(["LICENSE"])
 py_library(
     name = "python",
     srcs = ["__init__.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         "//tensorflow_model_optimization/python/core",
     ],

tensorflow_model_optimization/python/core/BUILD

@@ -7,7 +7,7 @@ exports_files(["LICENSE"])
 py_library(
     name = "core",
     srcs = ["__init__.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":version",
         "//tensorflow_model_optimization/python/core/clustering",
@@ -21,5 +21,5 @@ py_library(
 py_library(
     name = "version",
     srcs = ["version.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
 )

tensorflow_model_optimization/python/core/api/BUILD

@@ -16,7 +16,7 @@ py_library(
         "sparsity/__init__.py",
         "sparsity/keras/__init__.py",
     ],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         "//tensorflow_model_optimization/python/core/clustering/keras",
         "//tensorflow_model_optimization/python/core/quantization/keras",

tensorflow_model_optimization/python/core/clustering/BUILD

@@ -5,6 +5,6 @@ licenses(["notice"])  # Apache 2.0
 py_library(
     name = "clustering",
     srcs = ["__init__.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = ["//tensorflow_model_optimization/python/core/clustering/keras"],
 )

tensorflow_model_optimization/python/core/clustering/keras/BUILD

@@ -11,7 +11,7 @@ py_library(
     srcs = [
         "__init__.py",
     ],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         ":cluster",
     ],
@@ -20,7 +20,7 @@ py_library(
 py_library(
     name = "cluster",
     srcs = ["cluster.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
@@ -33,14 +33,14 @@ py_library(
 py_library(
     name = "cluster_config",
     srcs = ["cluster_config.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "clustering_registry",
     srcs = ["clustering_registry.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":clusterable_layer",
@@ -50,14 +50,14 @@ py_library(
 py_library(
     name = "clusterable_layer",
     srcs = ["clusterable_layer.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "clustering_centroids",
     srcs = ["clustering_centroids.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
@@ -67,7 +67,7 @@ py_library(
 py_library(
     name = "cluster_wrapper",
     srcs = ["cluster_wrapper.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",

tensorflow_model_optimization/python/core/internal/BUILD

@@ -5,7 +5,7 @@ licenses(["notice"])  # Apache 2.0
 py_library(
     name = "internal",
     srcs = ["__init__.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         "//tensorflow_model_optimization/python/core/internal/tensor_encoding",
     ],

tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD

@@ -5,7 +5,7 @@ licenses(["notice"])  # Apache 2.0
 py_library(
     name = "tensor_encoding",
     srcs = ["__init__.py"],
-    srcs_version = "PY2AND3",
+    srcs_version = "PY3",
     deps = [
         "//tensorflow_model_optimization/python/core/internal/tensor_encoding/core",
         "//tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders",

tensorflow_model_optimization/python/core/internal/tensor_encoding/core/BUILD

@@ -31,6 +31,7 @@ py_test(
     name = "core_encoder_test",
     size = "small",
     srcs = ["core_encoder_test.py"],
+    python_version = "PY3",
     deps = [
         ":core_encoder",
         ":encoding_stage",
@@ -54,6 +55,7 @@ py_test(
     name = "encoding_stage_test",
     size = "small",
     srcs = ["encoding_stage_test.py"],
+    python_version = "PY3",
     deps = [
         ":encoding_stage",
         # absl/testing:parameterized dep1,
@@ -78,6 +80,7 @@ py_test(
     name = "gather_encoder_test",
     size = "small",
     srcs = ["gather_encoder_test.py"],
+    python_version = "PY3",
     deps = [
         ":core_encoder",
         ":encoding_stage",
@@ -104,6 +107,7 @@ py_test(
     name = "simple_encoder_test",
     size = "small",
     srcs = ["simple_encoder_test.py"],
+    python_version = "PY3",
     deps = [
         ":core_encoder",
         ":simple_encoder",

tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/BUILD

@@ -29,6 +29,7 @@ py_test(
     name = "common_encoders_test",
     size = "small",
     srcs = ["common_encoders_test.py"],
+    python_version = "PY3",
     deps = [
         ":common_encoders",
         # absl/testing:parameterized dep1,