Fix image segmentation tool test (#23306)

huggingface · May 11, 2023 · df98769 · df98769
1 parent 3652e16
commit df98769
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/tools/test_image_segmentation.py b/tests/tools/test_image_segmentation.py
@@ -44,10 +44,10 @@ def test_exact_match_arg_remote(self):
 
     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))