Fix image segmentation tool test (#23306)

4eea25b4 · Sylvain Gugger · GitHub · 662751b4 · 4eea25b4
Unverified Commit 4eea25b4 authored May 11, 2023 by Sylvain Gugger Committed by GitHub May 11, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/tools/test_image_segmentation.py tests/tools/test_image_segmentation.py +2 -2

No files found.
--- a/tests/tools/test_image_segmentation.py
+++ b/tests/tools/test_image_segmentation.py
@@ -44,10 +44,10 @@ class ImageSegmentationToolTester(unittest.TestCase, ToolTesterMixin):

    def test_exact_match_kwarg(self):
        image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512))
-        result = self.tool(image=image, prompt="cat")
+        result = self.tool(image=image, label="cat")
        self.assertTrue(isinstance(result, Image.Image))

    def test_exact_match_kwarg_remote(self):
        image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512))
-        result = self.remote_tool(image=image, prompt="cat")
+        result = self.remote_tool(image=image, label="cat")
        self.assertTrue(isinstance(result, Image.Image))