From 4eea25b44530198c16c048096c791a2f749ebf58 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Thu, 11 May 2023 14:38:11 -0400 Subject: [PATCH] Fix image segmentation tool test (#23306) --- tests/tools/test_image_segmentation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tools/test_image_segmentation.py b/tests/tools/test_image_segmentation.py index b71d847299..8a741b501b 100644 --- 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))