diff --git a/tensorflow_probability/python/internal/BUILD b/tensorflow_probability/python/internal/BUILD index 2182c40599..ea5ed5b2f8 100644 --- a/tensorflow_probability/python/internal/BUILD +++ b/tensorflow_probability/python/internal/BUILD @@ -251,6 +251,7 @@ multi_substrate_py_library( multi_substrate_py_test( name = "custom_gradient_test", srcs = ["custom_gradient_test.py"], + numpy_tags = ["notap"], deps = [ ":custom_gradient", # tensorflow dep, @@ -468,6 +469,7 @@ multi_substrate_py_test( name = "numerics_testing_test", size = "small", srcs = ["numerics_testing_test.py"], + numpy_tags = ["notap"], deps = [ ":numerics_testing", # tensorflow dep,