Skip to content

Commit 67c936d

Browse files
committed
Fix test_vision_manual.py collected by pytest
Rename test_model_with_image to run_model_with_image so pytest does not collect it as a test function. It is a helper called from main() when the script is run directly.
1 parent ce1855c commit 67c936d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

libs/oci/tests/integration_tests/chat_models/test_vision_manual.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def create_solid_color_image(color: str = "red", size: tuple = (100, 100)) -> by
102102
return buffer.getvalue()
103103

104104

105-
def test_model_with_image(
105+
def run_model_with_image(
106106
model_id: str, image_bytes: bytes, prompt: str, image_desc: str
107107
) -> dict:
108108
"""Test a vision model with an image."""
@@ -164,7 +164,7 @@ def main():
164164
image_bytes = create_solid_color_image(color)
165165

166166
for model_id in VISION_MODELS:
167-
result = test_model_with_image(
167+
result = run_model_with_image(
168168
model_id,
169169
image_bytes,
170170
f"What color is this image? One word answer.",

0 commit comments

Comments
 (0)