Fix dependencies for core/clustering, core/common, and core/sparsity

PiperOrigin-RevId: 370621468

Author: teijeong

tensorflow_model_optimization/python/core/clustering/BUILD

@@ -1,10 +1,14 @@
+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 = "clustering",
     srcs = ["__init__.py"],
     srcs_version = "PY3",
-    deps = ["//tensorflow_model_optimization/python/core/clustering/keras"],
+    deps = [
+        "//tensorflow_model_optimization/python/core/clustering/keras",  # buildcleaner: keep
+    ],
 )

tensorflow_model_optimization/python/core/clustering/keras/BUILD

@@ -1,19 +1,25 @@
+# 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:__subpackages__",
 ])
 
 licenses(["notice"])
 
-py_library(
+py_strict_library(
     name = "keras",
     srcs = [
         "__init__.py",
     ],
     srcs_version = "PY3",
     deps = [
-        ":cluster",
-        ":clustering_callbacks",
-        "//tensorflow_model_optimization/python/core/clustering/keras/experimental",
+        ":cluster",  # buildcleaner: keep
+        ":clustering_callbacks",  # buildcleaner: keep
+        "//tensorflow_model_optimization/python/core/clustering/keras/experimental",  # buildcleaner: keep
     ],
 )
 
@@ -23,14 +29,13 @@ py_library(
     srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
-        ":cluster_config",
         ":cluster_wrapper",
         ":clustering_centroids",
-        ":clustering_registry",
+        # tensorflow dep1,
     ],
 )
 
-py_library(
+py_strict_library(
     name = "cluster_config",
     srcs = ["cluster_config.py"],
     srcs_version = "PY3",
@@ -45,23 +50,30 @@ py_library(
     deps = [
         ":clusterable_layer",
         ":clustering_algorithm",
+        # tensorflow dep1,
     ],
 )
 
-py_library(
+py_strict_library(
     name = "clusterable_layer",
     srcs = ["clusterable_layer.py"],
     srcs_version = "PY3",
     visibility = ["//visibility:public"],
+    deps = [
+        # six dep1,
+    ],
 )
 
-py_library(
+py_strict_library(
     name = "clustering_centroids",
     srcs = ["clustering_centroids.py"],
     srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
+        # six dep1,
+        # tensorflow dep1,
+        # python:clustering_ops tensorflow dep2,
     ],
 )
 
@@ -72,15 +84,20 @@ py_library(
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
+        ":clusterable_layer",
+        ":clustering_centroids",
+        ":clustering_registry",
+        # tensorflow dep1,
     ],
 )
 
-py_library(
+py_strict_library(
     name = "clustering_algorithm",
     srcs = ["clustering_algorithm.py"],
     srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
+        # six dep1,
         # tensorflow dep1,
         "//tensorflow_model_optimization/python/core/clustering/keras:cluster_config",
     ],
@@ -105,6 +122,11 @@ py_test(
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
+        ":cluster_config",
+        ":cluster_wrapper",
+        ":clusterable_layer",
+        ":clustering_registry",
+        # absl/testing:parameterized dep1,
         # tensorflow dep1,
         "//tensorflow_model_optimization/python/core/clustering/keras/experimental:cluster",
     ],
@@ -117,37 +139,40 @@ py_test(
     python_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
+        ":cluster_config",
         ":clustering_centroids",
+        # absl/testing:parameterized dep1,
         # tensorflow dep1,
     ],
 )
 
-py_test(
+py_strict_test(
     name = "cluster_wrapper_test",
     size = "medium",
     srcs = ["cluster_wrapper_test.py"],
     python_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
+        ":cluster_config",
         ":cluster_wrapper",
         ":clusterable_layer",
-        ":clustering_centroids",
-        ":clustering_registry",
+        # absl/testing:parameterized dep1,
         # tensorflow dep1,
     ],
 )
 
-py_test(
+py_strict_test(
     name = "clustering_registry_test",
     size = "medium",
     srcs = ["clustering_registry_test.py"],
     python_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
-        ":cluster_wrapper",
+        ":cluster_config",
         ":clusterable_layer",
         ":clustering_registry",
+        # absl/testing:parameterized dep1,
         # tensorflow dep1,
     ],
 )
@@ -160,42 +185,52 @@ py_test(
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
+        ":cluster_config",
+        # absl/testing:parameterized dep1,
+        # numpy dep1,
         # tensorflow dep1,
         "//tensorflow_model_optimization/python/core/clustering/keras/experimental:cluster",
-        "//tensorflow_model_optimization/python/core/keras:compat",
     ],
 )
 
-py_test(
+py_strict_test(
     name = "cluster_distributed_test",
     srcs = ["cluster_distributed_test.py"],
     python_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
+        ":cluster_config",
+        ":cluster_wrapper",
+        # absl/testing:parameterized dep1,
+        # numpy dep1,
         # tensorflow dep1,
         "//tensorflow_model_optimization/python/core/keras:test_utils",
     ],
 )
 
-py_test(
+py_strict_test(
     name = "mnist_clusterable_layer_test",
     srcs = ["mnist_clusterable_layer_test.py"],
     python_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
+        ":cluster_config",
+        ":clusterable_layer",
+        ":clustering_algorithm",
         # tensorflow dep1,
     ],
 )
 
-py_test(
+py_strict_test(
     name = "mnist_clustering_test",
     srcs = ["mnist_clustering_test.py"],
     python_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
+        ":cluster_config",
         # tensorflow dep1,
     ],
 )

tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD

@@ -1,21 +1,23 @@
+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 = "experimental",
     srcs = [
         "__init__.py",
     ],
     srcs_version = "PY3",
     deps = [
-        ":cluster",
+        ":cluster",  # buildcleaner: keep
     ],
 )
 
-py_library(
+py_strict_library(
     name = "cluster",
     srcs = ["cluster.py"],
     srcs_version = "PY3",

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

@@ -1,8 +1,13 @@
-load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "pytype_library")
+# 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")
+load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "pytype_strict_library")
 
 licenses(["notice"])
 
-pytype_library(
+pytype_strict_library(
     name = "algorithm",
     srcs = ["algorithm.py"],
     srcs_version = "PY3",
@@ -15,7 +20,7 @@ pytype_library(
     ],
 )
 
-py_library(
+py_strict_library(
     name = "schedules",
     srcs = ["schedules.py"],
     srcs_version = "PY3ONLY",
@@ -24,16 +29,14 @@ py_library(
     ],
 )
 
-py_test(
+py_strict_test(
     name = "schedules_test",
     srcs = [
         "schedules_test.py",
     ],
     python_version = "PY3",
     deps = [
         ":schedules",
-        # absl/testing:parameterized dep1,
-        # numpy dep1,
         # tensorflow dep1,
     ],
 )