From ad54e02d41f90b7de685f992ba1bd1fcf51412f9 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Fri, 8 Nov 2024 09:08:10 -0800 Subject: [PATCH] In jnp.reshape, use `shape` rather than deprecated `newshape`. The `newshape` parameter was deprecated in JAX v0.4.28, and will soon be removed. PiperOrigin-RevId: 694522820 --- dm_pix/_src/augment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dm_pix/_src/augment.py b/dm_pix/_src/augment.py index e4becce..0de0cb4 100644 --- a/dm_pix/_src/augment.py +++ b/dm_pix/_src/augment.py @@ -715,7 +715,7 @@ def affine_transform( # Alter coordinates to account for offset. offset = jnp.full((3,), fill_value=offset) - coordinates += jnp.reshape(a=offset, newshape=(*offset.shape, 1, 1, 1)) + coordinates += jnp.reshape(offset, (*offset.shape, 1, 1, 1)) interpolate_function = _get_interpolate_function( mode=mode,