Spaces:
Sleeping
Sleeping
File size: 505 Bytes
170d9a9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import pytest
def pytest_addoption(parser):
parser.addoption(
"--model-name",
action="store",
default="gemini-1.5-flash",
help="Model name to use for evaluation",
)
def pytest_configure(config):
# Add model_name to pytest namespace for access in tests
config.addinivalue_line(
"markers", "model_name: mark test to run with specific model name"
)
@pytest.fixture
def model_name(request):
return request.config.getoption("--model-name")
|