From 5807e1d3f4fcdeeef72a5b533a9cb73c5b353f5e Mon Sep 17 00:00:00 2001 From: Wassim Kabalan Date: Sun, 8 Jun 2025 15:37:06 +0200 Subject: [PATCH] format --- jaxpm/painting_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/jaxpm/painting_utils.py b/jaxpm/painting_utils.py index a27aed5..bc9cf4b 100644 --- a/jaxpm/painting_utils.py +++ b/jaxpm/painting_utils.py @@ -30,8 +30,7 @@ def enmesh(base_indices, displacements, cell_size, base_shape, offset, """Multilinear enmeshing.""" base_indices = jnp.asarray(base_indices) displacements = jnp.asarray(displacements) - cell_size = jnp.array( - cell_size, dtype=displacements.dtype) + cell_size = jnp.array(cell_size, dtype=displacements.dtype) if base_shape is not None: base_shape = jnp.array(base_shape, dtype=base_indices.dtype) offset = offset.astype(base_indices.dtype)