OSS cleanup: improve copybara process to keep strict rules.

This reduces technical debt from managing internal `strict` python rules. Added placeholder rules until bazel supports them.

PiperOrigin-RevId: 370019453

Author: teijeong

tensorflow_model_optimization/python/core/common/keras/compression/BUILD

@@ -1,6 +1,8 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "pytype_library")
+
 licenses(["notice"])
 
-py_library(
+pytype_library(
     name = "algorithm",
     srcs = ["algorithm.py"],
     srcs_version = "PY3",

tensorflow_model_optimization/python/core/common/keras/compression/algorithms/BUILD

@@ -1,8 +1,10 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "pytype_library")
+
 package(default_visibility = ["//visibility:private"])
 
 licenses(["notice"])
 
-py_library(
+pytype_library(
     name = "same_training_and_inference",
     srcs = ["same_training_and_inference.py"],
     srcs_version = "PY3",
@@ -24,7 +26,7 @@ py_test(
     ],
 )
 
-py_library(
+pytype_library(
     name = "different_training_and_inference",
     srcs = ["different_training_and_inference.py"],
     srcs_version = "PY3",
@@ -46,7 +48,7 @@ py_test(
     ],
 )
 
-py_library(
+pytype_library(
     name = "bias_only",
     srcs = ["bias_only.py"],
     srcs_version = "PY3",
@@ -91,7 +93,7 @@ py_test(
     ],
 )
 
-py_library(
+pytype_library(
     name = "periodical_update_and_scheduling",
     srcs = ["periodical_update_and_scheduling.py"],
     srcs_version = "PY3",

tensorflow_model_optimization/python/core/common/keras/compression/internal/BUILD

@@ -1,10 +1,12 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "pytype_library")
+
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/common/keras/compression:__subpackages__",
 ])
 
 licenses(["notice"])
 
-py_library(
+pytype_library(
     name = "optimize",
     srcs = ["optimize.py"],
     srcs_version = "PY3",

tensorflow_model_optimization/python/core/internal/BUILD

@@ -1,8 +1,10 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
 package(default_visibility = ["//visibility:public"])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "internal",
     srcs = ["__init__.py"],
     srcs_version = "PY3",

tensorflow_model_optimization/python/core/internal/tensor_encoding/BUILD

@@ -1,8 +1,10 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
 package(default_visibility = ["//visibility:public"])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "tensor_encoding",
     srcs = ["__init__.py"],
     srcs_version = "PY3",

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

@@ -1,10 +1,16 @@
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
+
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
 ])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "core",
     srcs = ["__init__.py"],
     visibility = ["//visibility:public"],
@@ -16,7 +22,7 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "core_encoder",
     srcs = ["core_encoder.py"],
     deps = [
@@ -27,7 +33,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "core_encoder_test",
     size = "small",
     srcs = ["core_encoder_test.py"],
@@ -41,7 +47,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "encoding_stage",
     srcs = ["encoding_stage.py"],
     deps = [
@@ -51,7 +57,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "encoding_stage_test",
     size = "small",
     srcs = ["encoding_stage_test.py"],
@@ -66,7 +72,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "gather_encoder",
     srcs = ["gather_encoder.py"],
     deps = [
@@ -76,7 +82,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "gather_encoder_test",
     size = "small",
     srcs = ["gather_encoder_test.py"],
@@ -93,7 +99,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "simple_encoder",
     srcs = ["simple_encoder.py"],
     deps = [
@@ -103,7 +109,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "simple_encoder_test",
     size = "small",
     srcs = ["simple_encoder_test.py"],

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

@@ -1,10 +1,16 @@
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
+
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
 ])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "encoders",
     srcs = ["__init__.py"],
     visibility = ["//visibility:public"],
@@ -13,7 +19,7 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "common_encoders",
     srcs = ["common_encoders.py"],
     deps = [
@@ -25,7 +31,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "common_encoders_test",
     size = "small",
     srcs = ["common_encoders_test.py"],

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

@@ -1,10 +1,16 @@
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
+
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
 ])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "stages",
     srcs = ["__init__.py"],
     visibility = ["//visibility:public"],
@@ -14,7 +20,7 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "stages_impl",
     srcs = ["stages_impl.py"],
     deps = [
@@ -25,7 +31,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "stages_impl_test",
     size = "medium",
     srcs = ["stages_impl_test.py"],

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

@@ -1,10 +1,16 @@
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
+
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
 ])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "research",
     srcs = ["__init__.py"],
     visibility = ["//visibility:public"],
@@ -16,7 +22,7 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "clipping",
     srcs = ["clipping.py"],
     deps = [
@@ -25,7 +31,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "clipping_test",
     size = "small",
     srcs = ["clipping_test.py"],
@@ -39,7 +45,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "kashin",
     srcs = ["kashin.py"],
     deps = [
@@ -50,7 +56,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "kashin_test",
     size = "medium",
     srcs = ["kashin_test.py"],
@@ -64,7 +70,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "quantization",
     srcs = ["quantization.py"],
     deps = [
@@ -74,7 +80,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "quantization_test",
     size = "small",
     srcs = ["quantization_test.py"],
@@ -88,7 +94,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "misc",
     srcs = ["misc.py"],
     deps = [
@@ -97,7 +103,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "misc_test",
     size = "small",
     srcs = ["misc_test.py"],

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

@@ -1,10 +1,16 @@
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
+# buildifier: disable=same-origin-load
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_test")
+
 package(default_visibility = [
     "//tensorflow_model_optimization/python/core/internal/tensor_encoding:__subpackages__",
 ])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "testing",
     srcs = ["__init__.py"],
     visibility = ["//visibility:public"],
@@ -13,7 +19,7 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "test_utils",
     srcs = ["test_utils.py"],
     deps = [
@@ -26,7 +32,7 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "test_utils_test",
     size = "medium",
     srcs = ["test_utils_test.py"],