Skip to content

Commit

Permalink
add test for blip
Browse files Browse the repository at this point in the history
  • Loading branch information
HRashidi committed Dec 6, 2024
1 parent 01a6f08 commit 40b16bc
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aana/tests/deployments/test_hf_blip2_deployment.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ class TestHFBlip2Deployment:
"""Test HuggingFace BLIP2 deployment."""

@pytest.mark.asyncio
@pytest.mark.parametrize("image_name", ["Starry_Night.jpeg"])
@pytest.mark.parametrize("image_name", ["Starry_Night.jpeg", "Seated_Man.png"])
async def test_methods(self, setup_deployment, image_name):
"""Test HuggingFace BLIP2 methods."""
deployment_name, handle_name, _ = setup_deployment
Expand Down
Binary file added aana/tests/files/images/Seated_Man.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 40b16bc

Please sign in to comment.