Automated Code Change

tensorflow_model_optimization/python/core/clustering/keras/BUILD

@@ -1,4 +1,3 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
 # Placeholder: load py_test
 
@@ -137,7 +136,7 @@ py_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "clustering_centroids_test",
     size = "medium",
     srcs = ["clustering_centroids_test.py"],
@@ -153,7 +152,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "cluster_wrapper_test",
     size = "medium",
     srcs = ["cluster_wrapper_test.py"],
@@ -170,7 +169,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "clustering_registry_test",
     size = "medium",
     srcs = ["clustering_registry_test.py"],
@@ -204,7 +203,7 @@ py_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "cluster_distributed_test",
     srcs = ["cluster_distributed_test.py"],
     visibility = ["//visibility:public"],
@@ -222,7 +221,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "mnist_clusterable_layer_test",
     srcs = ["mnist_clusterable_layer_test.py"],
     tags = ["requires-net:external"],
@@ -238,7 +237,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "mnist_clustering_test",
     srcs = ["mnist_clustering_test.py"],
     tags = ["requires-net:external"],

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

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
@@ -32,7 +32,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "core_encoder_test",
     size = "small",
     srcs = ["core_encoder_test.py"],
@@ -56,7 +56,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "encoding_stage_test",
     size = "small",
     srcs = ["encoding_stage_test.py"],
@@ -82,7 +82,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "gather_encoder_test",
     size = "small",
     srcs = ["gather_encoder_test.py"],
@@ -110,7 +110,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "simple_encoder_test",
     size = "small",
     srcs = ["simple_encoder_test.py"],

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

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
@@ -30,7 +30,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "common_encoders_test",
     size = "small",
     srcs = ["common_encoders_test.py"],

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

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
@@ -30,7 +30,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "stages_impl_test",
     size = "medium",
     srcs = ["stages_impl_test.py"],

tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/BUILD

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
@@ -30,7 +30,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "clipping_test",
     size = "small",
     srcs = ["clipping_test.py"],
@@ -56,7 +56,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "kashin_test",
     size = "medium",
     srcs = ["kashin_test.py"],
@@ -81,7 +81,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantization_test",
     size = "small",
     srcs = ["quantization_test.py"],
@@ -105,7 +105,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "misc_test",
     size = "small",
     srcs = ["misc_test.py"],

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

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
@@ -31,7 +31,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "test_utils_test",
     size = "medium",
     srcs = ["test_utils_test.py"],

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

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
@@ -30,7 +30,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "py_utils_test",
     size = "small",
     srcs = ["py_utils_test.py"],
@@ -53,7 +53,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "tf_utils_test",
     size = "medium",
     srcs = ["tf_utils_test.py"],

tensorflow_model_optimization/python/core/keras/BUILD

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization:__subpackages__",
@@ -59,7 +59,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "metrics_test",
     srcs = ["metrics_test.py"],
     deps = [

tensorflow_model_optimization/python/core/quantization/keras/BUILD

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization:__subpackages__",
@@ -36,7 +36,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quant_ops_test",
     size = "small",
     srcs = ["quant_ops_test.py"],
@@ -64,7 +64,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantizers_test",
     srcs = [
         "quantizers_test.py",
@@ -134,7 +134,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_annotate_test",
     srcs = [
         "quantize_annotate_test.py",
@@ -165,7 +165,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_aware_activation_test",
     srcs = [
         "quantize_aware_activation_test.py",
@@ -199,7 +199,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_layer_test",
     srcs = [
         "quantize_layer_test.py",
@@ -233,7 +233,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_wrapper_test",
     srcs = [
         "quantize_wrapper_test.py",
@@ -275,7 +275,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_test",
     srcs = ["quantize_test.py"],
     visibility = ["//visibility:public"],
@@ -296,7 +296,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_integration_test",
     srcs = ["quantize_integration_test.py"],
     deps = [
@@ -313,7 +313,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_models_test",
     size = "enormous",
     srcs = ["quantize_models_test.py"],
@@ -329,7 +329,7 @@ py_strict_test(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "quantize_functional_test",
     size = "large",
     srcs = ["quantize_functional_test.py"],

tensorflow_model_optimization/python/core/quantization/keras/default_8bit/BUILD

@@ -1,4 +1,3 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
 # Placeholder: load py_test
 
@@ -113,7 +112,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "default_8bit_transforms_test",
     size = "large",
     srcs = [

tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/BUILD

@@ -1,4 +1,3 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
 # Placeholder: load py_test
 
@@ -124,7 +123,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "default_n_bit_transforms_test",
     size = "large",
     srcs = [

tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/BUILD

@@ -1,5 +1,5 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
 # Placeholder: load py_library
+# Placeholder: load py_test
 
 package(default_visibility = [
     "//tensorflow_model_optimization:__subpackages__",
@@ -28,7 +28,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "transforms_test",
     srcs = [
         "transforms_test.py",
@@ -55,7 +55,7 @@ py_library(
     ],
 )
 
-py_strict_test(
+py_test(
     name = "model_transformer_test",
     srcs = [
         "model_transformer_test.py",