From 7bb0662a756da274177daf6d831353e19d86360c Mon Sep 17 00:00:00 2001 From: phawkins Date: Mon, 9 Dec 2024 09:47:09 -0800 Subject: [PATCH] [numpy] Fix test failures under NumPy 2.2. PiperOrigin-RevId: 704320367 --- tensorflow_probability/python/math/integration.py | 2 +- tensorflow_probability/python/sts/internal/seasonality_util.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_probability/python/math/integration.py b/tensorflow_probability/python/math/integration.py index 34ea61ad45..df0e34c758 100644 --- a/tensorflow_probability/python/math/integration.py +++ b/tensorflow_probability/python/math/integration.py @@ -69,7 +69,7 @@ def trapz( if dx is None: dx = 1. dx = tf.convert_to_tensor(dx, dtype=dtype, name='dx') - if ps.shape(dx): + if ps.shape(dx).size > 0: raise ValueError('Expected dx to be a scalar, got dx={}'.format(dx)) elem_sum = tf.reduce_sum(y, axis=axis) elem_sum -= 0.5 * tf.reduce_sum( diff --git a/tensorflow_probability/python/sts/internal/seasonality_util.py b/tensorflow_probability/python/sts/internal/seasonality_util.py index 042e5a52ad..776fa19a21 100644 --- a/tensorflow_probability/python/sts/internal/seasonality_util.py +++ b/tensorflow_probability/python/sts/internal/seasonality_util.py @@ -178,7 +178,7 @@ def build_fixed_effects(num_steps, else: design_matrix = ps.ones([num_steps, 1], dtype=dtype) - if covariates: + if covariates is not None: design_matrix = ps.concat( [design_matrix] + [tf.convert_to_tensor(x)[..., :num_steps, :] for x in covariates],