Skip to content

Commit b35e008

Browse files
committed
Suppress Cholesky failures in kernel_properties_test.
1 parent 7bee0d0 commit b35e008

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tensorflow_probability/python/math/psd_kernels/psd_kernel_properties_test.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,8 +143,9 @@ def testCompositeTensor(self, kernel_name, data):
143143
example_ndims=1,
144144
feature_dim=2,
145145
feature_ndims=1)))
146-
with tfp_hps.no_tf_rank_errors():
147-
diag = kernel.apply(xs, xs, example_ndims=1)
146+
with tfp_hps.no_cholesky_decomposition_errors():
147+
with tfp_hps.no_tf_rank_errors():
148+
diag = kernel.apply(xs, xs, example_ndims=1)
148149

149150
# Test flatten/unflatten.
150151
flat = tf.nest.flatten(kernel, expand_composites=True)

0 commit comments

Comments
 (0)