Fix dependencies for python/, core/quantization/

PiperOrigin-RevId: 370621567

Author: teijeong

tensorflow_model_optimization/python/core/quantization/BUILD

@@ -1,10 +1,14 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
 package(default_visibility = ["//tensorflow_model_optimization:__subpackages__"])
 
 licenses(["notice"])  # Apache 2.0
 
-py_library(
+py_strict_library(
     name = "quantization",
     srcs = ["__init__.py"],
     srcs_version = "PY3",
-    deps = ["//tensorflow_model_optimization/python/core/quantization/keras"],
+    deps = [
+        "//tensorflow_model_optimization/python/core/quantization/keras",  # buildcleaner: keep
+    ],
 )

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/BUILD

@@ -1,10 +1,12 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
 package(default_visibility = [
     "//tensorflow_model_optimization:__subpackages__",
 ])
 
 licenses(["notice"])
 
-py_library(
+py_strict_library(
     name = "cluster_preserve",
     srcs = [
         "__init__.py",
@@ -15,15 +17,14 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "cluster_utils",
     srcs = [
         "cluster_utils.py",
     ],
     srcs_version = "PY3",
     deps = [
         # tensorflow dep1,
-        "//tensorflow_model_optimization/python/core/clustering/keras:clustering_registry",
     ],
 )
 
@@ -36,6 +37,9 @@ py_library(
     deps = [
         ":cluster_utils",
         # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/clustering/keras:clustering_registry",
+        "//tensorflow_model_optimization/python/core/quantization/keras:quant_ops",
+        "//tensorflow_model_optimization/python/core/quantization/keras:quantizers",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_registry",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantizers",
     ],
@@ -50,13 +54,15 @@ py_test(
     visibility = ["//visibility:private"],
     deps = [
         ":cluster_preserve_quantize_registry",
+        # absl/testing:parameterized dep1,
         # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/clustering/keras:clustering_registry",
         "//tensorflow_model_optimization/python/core/quantization/keras:quantize_config",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_registry",
     ],
 )
 
-py_library(
+py_strict_library(
     name = "default_8bit_cluster_preserve_quantize_scheme",
     srcs = [
         "default_8bit_cluster_preserve_quantize_scheme.py",

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve/BUILD

@@ -1,10 +1,12 @@
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_library")
+
 package(default_visibility = [
     "//tensorflow_model_optimization:__subpackages__",
 ])
 
 licenses(["notice"])
 
-py_library(
+py_strict_library(
     name = "prune_preserve",
     srcs = [
         "__init__.py",
@@ -15,14 +17,16 @@ py_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "prune_preserve_quantize_registry",
     srcs = [
         "prune_preserve_quantize_registry.py",
     ],
     srcs_version = "PY3",
     deps = [
         # tensorflow dep1,
+        "//tensorflow_model_optimization/python/core/quantization/keras:quant_ops",
+        "//tensorflow_model_optimization/python/core/quantization/keras:quantizers",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_registry",
         "//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantizers",
     ],
@@ -43,7 +47,7 @@ py_test(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "default_8bit_prune_preserve_quantize_scheme",
     srcs = [
         "default_8bit_prune_preserve_quantize_scheme.py",