Skip to content

Commit ea6efb6

Browse files
srvasudetensorflower-gardener
authored andcommitted
Change python tests to use specific imports to improve build / test times.
- Leaves layers/ untouched. PiperOrigin-RevId: 472862069
1 parent f7da474 commit ea6efb6

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

57 files changed

+3663
-2595
lines changed

tensorflow_probability/python/bijectors/BUILD

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1059,12 +1059,27 @@ multi_substrate_py_test(
10591059
srcs = ["bijector_test.py"],
10601060
numpy_tags = ["notap"],
10611061
deps = [
1062-
":bijectors",
1062+
":bijector",
1063+
":exp",
1064+
":invert",
1065+
":joint_map",
1066+
":matrix_inverse_tril",
1067+
":power",
1068+
":reshape",
1069+
":restructure",
1070+
":scale",
1071+
":scale_matvec_diag",
1072+
":shift",
1073+
":sigmoid",
1074+
":sinh_arcsinh",
1075+
":softmax_centered",
1076+
":softplus",
1077+
":split",
10631078
# mock dep,
10641079
# numpy dep,
10651080
# tensorflow dep,
1066-
"//tensorflow_probability",
1067-
"//tensorflow_probability/python/distributions",
1081+
"//tensorflow_probability/python/distributions:distribution",
1082+
"//tensorflow_probability/python/distributions:transformed_distribution",
10681083
"//tensorflow_probability/python/internal:cache_util",
10691084
"//tensorflow_probability/python/internal:parameter_properties",
10701085
"//tensorflow_probability/python/internal:tensor_util",

0 commit comments

Comments
 (0)