diff --git a/tests/pytest/test_pytest_input_messages.py b/tests/pytest/test_pytest_input_messages.py index 9a35fc59..c601c74b 100644 --- a/tests/pytest/test_pytest_input_messages.py +++ b/tests/pytest/test_pytest_input_messages.py @@ -1,9 +1,11 @@ from typing import List +import pytest from eval_protocol.models import EvaluationRow, Message from eval_protocol.pytest import SingleTurnRolloutProcessor, evaluation_test +@pytest.mark.parametrize("completion_params", [{"model": "fireworks_ai/accounts/fireworks/models/gpt-oss-120b"}]) @evaluation_test( input_messages=[ [ @@ -12,7 +14,6 @@ ] ] ], - completion_params=[{"model": "fireworks_ai/accounts/fireworks/models/gpt-oss-120b"}], rollout_processor=SingleTurnRolloutProcessor(), mode="all", )