Fix test for numpy 2

keras-team · Sep 30, 2024 · 443284f · 443284f
1 parent d3671cf
commit 443284f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/src/ops/image_test.py b/keras/src/ops/image_test.py
@@ -351,7 +351,7 @@ def _compute_affine_transform_coordinates(image, transform):
     # transform the indices
     coordinates = np.einsum("Bhwij, Bjk -> Bhwik", indices, transform)
     coordinates = np.moveaxis(coordinates, source=-1, destination=1)
-    coordinates += np.reshape(a=offset, newshape=(*offset.shape, 1, 1, 1))
+    coordinates += np.reshape(offset, newshape=(*offset.shape, 1, 1, 1))
     if need_squeeze:
         coordinates = np.squeeze(coordinates, axis=0)
     return coordinates