From 3adb838d4bee35a8846d0382ee65234a45b64d12 Mon Sep 17 00:00:00 2001 From: thomaswc Date: Wed, 6 Mar 2024 10:38:56 -0800 Subject: [PATCH] s/total_shape/concentration_shape/ PiperOrigin-RevId: 613268316 --- .../python/distributions/hypothesis_testlib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_probability/python/distributions/hypothesis_testlib.py b/tensorflow_probability/python/distributions/hypothesis_testlib.py index 52eeaaa253..0f8b446ab6 100644 --- a/tensorflow_probability/python/distributions/hypothesis_testlib.py +++ b/tensorflow_probability/python/distributions/hypothesis_testlib.py @@ -274,7 +274,7 @@ def fix_bates(d): 'tailweight': tfp_hps.softplus_plus_eps(), 'temperature': tfp_hps.softplus_plus_eps(), 'total_count': lambda x: tf.floor(tf.sigmoid(x / 100) * 100) + 1, - 'total_shape': tfp_hps.shapes(min_ndims=1, min_lastdimsize=2), + 'concentration_shape': tfp_hps.shapes(min_ndims=1, min_lastdimsize=2), 'Bates': fix_bates, 'Bernoulli': lambda d: dict(d, dtype=tf.float32), 'CholeskyLKJ': fix_lkj,