Suppress TensorFlow rank errors when batch shape testing.

PiperOrigin-RevId: 387639055

Author: axch

tensorflow_probability/python/distributions/distribution_properties_test.py

@@ -373,7 +373,8 @@ def testInferredBatchShapeMatchesTrueBatchShape(self, dist_name, data):
     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()))
+      with tfp_hps.no_tf_rank_errors():
+        lp = dist.log_prob(dist.sample(seed=test_util.test_seed()))
 
     self.assertAllEqual(dist.batch_shape_tensor(), tf.shape(lp))
     self.assertAllEqual(dist.batch_shape, tf.shape(lp))