From 90d36dc0dfdafeb0dedb9dab1172b1af9a5a8178 Mon Sep 17 00:00:00 2001 From: Francois Chollet Date: Wed, 11 Dec 2024 19:28:50 -0800 Subject: [PATCH] Fix torch gpu ci --- .../image_preprocessing/bounding_boxes/converters_test.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/converters_test.py b/keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/converters_test.py index 6cb18e56950..79ec220f58b 100644 --- a/keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/converters_test.py +++ b/keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/converters_test.py @@ -3,6 +3,7 @@ import numpy as np from absl.testing import parameterized +from keras.src import ops from keras.src import testing from keras.src.layers.preprocessing.image_preprocessing.bounding_boxes.converters import ( # noqa: E501 affine_transform, @@ -139,4 +140,5 @@ def test_affine_identity(self): height=self.height, width=self.width, ) + transformed_boxes = ops.convert_to_numpy(transformed_boxes) self.assertAllClose(self.boxes["xyxy"], transformed_boxes)