Hypothesis: Also silence cholesky errors when constructing the distribution.

PiperOrigin-RevId: 387202839

Author: SiegeLordEx

tensorflow_probability/python/distributions/distribution_properties_test.py

@@ -370,9 +370,9 @@ def testCanConstructAndSampleDistribution(self, data):
   @hp.given(hps.data())
   @tfp_hps.tfp_hp_settings()
   def testInferredBatchShapeMatchesTrueBatchShape(self, dist_name, data):
-    dist = data.draw(
-        dhps.distributions(dist_name=dist_name, validate_args=False))
     with tfp_hps.no_cholesky_decomposition_errors():
+      dist = data.draw(
+          dhps.distributions(dist_name=dist_name, validate_args=False))
       lp = dist.log_prob(dist.sample(seed=test_util.test_seed()))
 
     self.assertAllEqual(dist.batch_shape_tensor(), tf.shape(lp))