Automated Code Change

PiperOrigin-RevId: 861669255

Author: h-joo

tensorflow_probability/python/bijectors/BUILD

@@ -464,6 +464,7 @@ multi_substrate_py_library(
 py_library(
     name = "glow",
     srcs = ["glow.py"],
+    strict_deps = False,
     deps = [
         ":bijector",
         ":blockwise",

tensorflow_probability/python/debugging/BUILD

@@ -28,6 +28,7 @@ licenses(["notice"])
 py_library(
     name = "debugging",
     srcs = ["__init__.py"],
+    strict_deps = False,
     deps = [
         "//tensorflow_probability/python/debugging/benchmarking",
     ],

tensorflow_probability/python/debugging/benchmarking/BUILD

@@ -28,6 +28,7 @@ licenses(["notice"])
 py_library(
     name = "benchmarking",
     srcs = ["__init__.py"],
+    strict_deps = False,
     deps = [
         ":benchmark_tf_function",
     ],
@@ -36,6 +37,7 @@ py_library(
 py_library(
     name = "benchmark_tf_function",
     srcs = ["benchmark_tf_function.py"],
+    strict_deps = False,
     deps = [
         # tensorflow dep,
     ],

tensorflow_probability/python/distributions/BUILD

@@ -4871,6 +4871,7 @@ py_test(
 py_library(
     name = "jax_transformation_testlib",
     testonly = 1,
+    strict_deps = False,
     deps = [
         ":hypothesis_testlib.jax",
         # absl/testing:parameterized dep,

tensorflow_probability/python/distributions/internal/BUILD

@@ -86,6 +86,7 @@ py_library(
     srcs = [
         "correlation_matrix_volumes_lib.py",
     ],
+    strict_deps = False,
     deps = [
         ":init",
         # numpy dep,

tensorflow_probability/python/experimental/auto_batching/BUILD

@@ -30,6 +30,7 @@ licenses(["notice"])
 py_library(
     name = "auto_batching",
     srcs = ["__init__.py"],
+    strict_deps = False,
     deps = [
         ":allocation_strategy",
         ":dsl",
@@ -49,6 +50,7 @@ py_library(
 py_library(
     name = "instructions",
     srcs = ["instructions.py"],
+    strict_deps = False,
     deps = [
         # numpy dep,
         # tensorflow dep,  # For pywrap_tensorflow.IsNamedtuple
@@ -60,6 +62,7 @@ py_library(
     srcs = [
         "numpy_backend.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
         # numpy dep,
@@ -71,6 +74,7 @@ py_library(
     srcs = [
         "tf_backend.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
         ":xla",
@@ -83,6 +87,7 @@ py_library(
     srcs = [
         "type_inference.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
         # absl/logging dep,
@@ -94,6 +99,7 @@ py_library(
     srcs = [
         "virtual_machine.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
     ],
@@ -104,6 +110,7 @@ py_library(
     srcs = [
         "liveness.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
     ],
@@ -114,6 +121,7 @@ py_library(
     srcs = [
         "lowering.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
         ":liveness",
@@ -125,6 +133,7 @@ py_library(
     srcs = [
         "stack_optimization.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
     ],
@@ -135,6 +144,7 @@ py_library(
     srcs = [
         "allocation_strategy.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
         ":liveness",
@@ -146,6 +156,7 @@ py_library(
     srcs = [
         "stackless.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
     ],
@@ -156,6 +167,7 @@ py_library(
     srcs = [
         "dsl.py",
     ],
+    strict_deps = False,
     deps = [
         ":instructions",
     ],
@@ -167,6 +179,7 @@ py_library(
         "frontend.py",
         "gast_util.py",
     ],
+    strict_deps = False,
     deps = [
         ":allocation_strategy",
         ":dsl",
@@ -187,6 +200,7 @@ py_library(
     srcs = [
         "backend_test_lib.py",
     ],
+    strict_deps = False,
     deps = [
         # tensorflow dep,
     ],
@@ -195,6 +209,7 @@ py_library(
 py_library(
     name = "test_programs",
     srcs = ["test_programs.py"],
+    strict_deps = False,
     deps = [
         ":instructions",
         # numpy dep,
@@ -204,6 +219,7 @@ py_library(
 py_library(
     name = "xla",
     srcs = ["xla.py"],
+    strict_deps = False,
     deps = [
         # tensorflow dep,
     ],

tensorflow_probability/python/experimental/fastgp/BUILD

@@ -30,6 +30,7 @@ package(
 py_library(
     name = "fastgp.jax",
     srcs = ["__init__.py"],
+    strict_deps = False,
     deps = [
         ":fast_gp",
         ":fast_gprm",
@@ -49,17 +50,20 @@ py_library(
 # tfp/python/experimental:experimental.
 py_library(
     name = "fastgp",
+    strict_deps = False,
     deps = [],
 )
 
 py_library(
     name = "fastgp.numpy",
+    strict_deps = False,
     deps = [],
 )
 
 py_library(
     name = "mbcg",
     srcs = ["mbcg.py"],
+    strict_deps = False,
     deps = [
     ],
 )
@@ -79,6 +83,7 @@ py_test(
 py_library(
     name = "fast_gp",
     srcs = ["fast_gp.py"],
+    strict_deps = False,
     deps = [
         ":fast_log_det",
         ":mbcg",
@@ -113,6 +118,7 @@ py_test(
 py_library(
     name = "fast_mtgp",
     srcs = ["fast_mtgp.py"],
+    strict_deps = False,
     deps = [
         ":fast_gp",
         ":fast_log_det",
@@ -146,6 +152,7 @@ py_test(
 py_library(
     name = "fast_gprm",
     srcs = ["fast_gprm.py"],
+    strict_deps = False,
     deps = [
         ":fast_gp",
         ":preconditioners",
@@ -174,6 +181,7 @@ py_test(
 py_library(
     name = "linalg",
     srcs = ["linalg.py"],
+    strict_deps = False,
     deps = [
         ":partial_lanczos",
         # jax/experimental:sparse dep,
@@ -197,6 +205,7 @@ py_test(
 py_library(
     name = "partial_lanczos",
     srcs = ["partial_lanczos.py"],
+    strict_deps = False,
     deps = [
         ":mbcg",
         # scipy dep,
@@ -220,6 +229,7 @@ py_test(
 py_library(
     name = "fast_log_det",
     srcs = ["fast_log_det.py"],
+    strict_deps = False,
     deps = [
         ":mbcg",
         ":partial_lanczos",
@@ -248,13 +258,15 @@ py_test(
 py_library(
     name = "linear_operator_sum",
     srcs = ["linear_operator_sum.py"],
+    strict_deps = False,
     deps = [
     ],
 )
 
 py_library(
     name = "preconditioners",
     srcs = ["preconditioners.py"],
+    strict_deps = False,
     deps = [
         ":linalg",
         ":linear_operator_sum",
@@ -279,6 +291,7 @@ py_test(
 py_library(
     name = "schur_complement",
     srcs = ["schur_complement.py"],
+    strict_deps = False,
     deps = [
         ":preconditioners",
         "//tensorflow_probability/python/bijectors:softplus.jax",

tensorflow_probability/python/experimental/linalg/BUILD

@@ -87,6 +87,7 @@ multi_substrate_py_test(
 py_library(
     name = "linear_operator_psd_kernel",
     srcs = ["linear_operator_psd_kernel.py"],
+    strict_deps = False,
     deps = [
         # numpy dep,
         # tensorflow dep,
@@ -158,6 +159,7 @@ multi_substrate_py_test(
 py_library(
     name = "no_pivot_ldl",
     srcs = ["no_pivot_ldl.py"],
+    strict_deps = False,
     deps = [
         # tensorflow dep,
     ],

tensorflow_probability/python/experimental/mcmc/BUILD

@@ -132,6 +132,7 @@ multi_substrate_py_test(
 py_library(
     name = "kernel_builder",
     srcs = ["kernel_builder.py"],
+    strict_deps = False,
     deps = [
         ":kernel_outputs",
         ":preconditioned_hmc",
@@ -173,6 +174,7 @@ py_test(
 py_library(
     name = "kernel_outputs",
     srcs = ["kernel_outputs.py"],
+    strict_deps = False,
     deps = [
         ":tracing_reducer",
         "//tensorflow_probability/python/internal:unnest",
@@ -374,6 +376,7 @@ multi_substrate_py_test(
 py_library(
     name = "progress_bar_reducer",
     srcs = ["progress_bar_reducer.py"],
+    strict_deps = False,
     deps = [
         ":reducer",
         # tensorflow dep,
@@ -426,6 +429,7 @@ multi_substrate_py_test(
 py_library(
     name = "nuts_autobatching",
     srcs = ["nuts_autobatching.py"],
+    strict_deps = False,
     deps = [
         # numpy dep,
         # tensorflow dep,
@@ -441,6 +445,7 @@ py_library(
     name = "nuts_autobatching_testlib",
     testonly = 1,
     srcs = ["nuts_autobatching_test.py"],
+    strict_deps = False,
     deps = [
         ":nuts_autobatching",
         # absl/testing:parameterized dep,
@@ -588,6 +593,7 @@ multi_substrate_py_test(
 py_library(
     name = "particle_filter_augmentation",
     srcs = ["particle_filter_augmentation.py"],
+    strict_deps = False,
     deps = [
         # numpy dep,
         # tensorflow dep,

tensorflow_probability/python/experimental/nn/BUILD

@@ -30,6 +30,7 @@ package(
 py_library(
     name = "nn",
     srcs = ["__init__.py"],
+    strict_deps = False,
     deps = [
         ":affine_layers",
         ":convolutional_layers",
@@ -47,6 +48,7 @@ py_library(
 py_library(
     name = "affine_layers",
     srcs = ["affine_layers.py"],
+    strict_deps = False,
     deps = [
         ":layers",
         ":variational_base",
@@ -82,6 +84,7 @@ py_test(
 py_library(
     name = "convolutional_layers",
     srcs = ["convolutional_layers.py"],
+    strict_deps = False,
     deps = [
         ":layers",
         ":variational_base",
@@ -115,6 +118,7 @@ py_test(
 py_library(
     name = "convolutional_layers_v2",
     srcs = ["convolutional_layers_v2.py"],
+    strict_deps = False,
     deps = [
         ":layers",
         ":variational_base",
@@ -149,6 +153,7 @@ py_test(
 py_library(
     name = "convolutional_transpose_layers",
     srcs = ["convolutional_transpose_layers.py"],
+    strict_deps = False,
     deps = [
         ":layers",
         ":variational_base",
@@ -180,6 +185,7 @@ py_test(
 py_library(
     name = "layers",
     srcs = ["layers.py"],
+    strict_deps = False,
     deps = [
         # tensorflow dep,
         "//tensorflow_probability/python/experimental/nn/util:utils",
@@ -203,6 +209,7 @@ py_test(
 py_library(
     name = "variational_base",
     srcs = ["variational_base.py"],
+    strict_deps = False,
     deps = [
         ":layers",
         # tensorflow dep,