Fix input_boxes shapes

Author: Rocketknight1

src/transformers/models/sam/processing_sam.py

@@ -223,7 +223,7 @@ def _check_and_preprocess_points(
                 input_points = input_points.numpy().tolist()
 
             if not isinstance(input_points, list) or not isinstance(input_points[0], list):
-                raise ValueError("Input points must be a list of list of floating integers.")
+                raise ValueError("Input points must be a list of list of floating points.")
             input_points = [np.array(input_point) for input_point in input_points]
         else:
             input_points = None
@@ -247,7 +247,7 @@ def _check_and_preprocess_points(
                 or not isinstance(input_boxes[0], list)
                 or not isinstance(input_boxes[0][0], list)
             ):
-                raise ValueError("Input boxes must be a list of list of list of floating integers.")
+                raise ValueError("Input boxes must be a list of list of list of floating points.")
             input_boxes = [np.array(box).astype(np.float32) for box in input_boxes]
         else:
             input_boxes = None

tests/models/sam/test_modeling_tf_sam.py

@@ -448,7 +448,7 @@ def test_inference_mask_generation_one_point_one_bb(self):
         processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 
         raw_image = prepare_image()
-        input_boxes = [[650, 900, 1000, 1250]]
+        input_boxes = [[[650, 900, 1000, 1250]]]
         input_points = [[[820, 1080]]]
 
         inputs = processor(images=raw_image, input_boxes=input_boxes, input_points=input_points, return_tensors="tf")
@@ -501,7 +501,7 @@ def test_inference_mask_generation_one_point_one_bb_zero(self):
         processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 
         raw_image = prepare_image()
-        input_boxes = [[620, 900, 1000, 1255]]
+        input_boxes = [[[620, 900, 1000, 1255]]]
         input_points = [[[820, 1080]]]
         labels = [[0]]