Skip to content

GitLab

Unverified Commit 4eea25b4 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix image segmentation tool test (#23306)

parent 662751b4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/tools/test_image_segmentation.py
View file @ 4eea25b4
...@@ -44,10 +44,10 @@ class ImageSegmentationToolTester(unittest.TestCase, ToolTesterMixin): ...@@ -44,10 +44,10 @@ class ImageSegmentationToolTester(unittest.TestCase, ToolTesterMixin):
def test_exact_match_kwarg(self): def test_exact_match_kwarg(self):
image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512)) 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)) self.assertTrue(isinstance(result, Image.Image))
def test_exact_match_kwarg_remote(self): def test_exact_match_kwarg_remote(self):
image = Image.open(Path(get_tests_dir("fixtures/tests_samples/COCO")) / "000000039769.png").resize((512, 512)) 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)) self.assertTrue(isinstance(result, Image.Image))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment