Fix torch gpu ci

fkouteib · Dec 12, 2024 · 90d36dc · 90d36dc
1 parent eec1fbd
commit 90d36dc
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -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)