Fix the prompt generator test

Author: constantinpape

test/test_prompt_generators.py

@@ -53,7 +53,7 @@ def test_point_prompt_generator(self):
             generator = PointAndBoxPromptGenerator(n_pos, n_neg, dilation_strength=4)
             for label_id in label_ids:
                 center, box = centers.get(label_id), boxes.get(label_id)
-                coords, point_labels, _, _ = generator(labels, label_id, center, box)
+                coords, point_labels, _, _ = generator(labels, label_id, box, center)
                 coords_ = (np.array([int(coo[0]) for coo in coords]),
                            np.array([int(coo[1]) for coo in coords]))
                 mask = labels == label_id
@@ -80,7 +80,7 @@ def test_box_prompt_generator(self):
 
         for label_id in label_ids:
             center, box_ = centers.get(label_id), boxes.get(label_id)
-            _, _, box, _ = generator(labels, label_id, center, box_)
+            _, _, box, _ = generator(labels, label_id, box_, center)
             coords = np.where(labels == label_id)
             expected_box = [coo.min() for coo in coords] + [coo.max() + 1 for coo in coords]
             self.assertEqual(expected_box, list(box[0]))