From 92be4e49f3c9d8f247e5d4dbb200654418e1d9d7 Mon Sep 17 00:00:00 2001 From: Dylan Huang Date: Tue, 23 Sep 2025 17:15:55 -0700 Subject: [PATCH] use decorator instead --- tests/pytest/test_pytest_input_messages.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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", )