use decorator instead (#212)

Dylan Huang · web-flow · commit aa7f2ab113fe · 2025-09-23T17:22:50.000-07:00
diff --git 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",
 )

Original file line number	Diff line number	Diff line change
`@@ -1,9 +1,11 @@`
`1`	`1`	`from typing import List`
`2`	`2`
	`3`	`+import pytest`
`3`	`4`	`from eval_protocol.models import EvaluationRow, Message`
`4`	`5`	`from eval_protocol.pytest import SingleTurnRolloutProcessor, evaluation_test`
`5`	`6`
`6`	`7`
	`8`	`+@pytest.mark.parametrize("completion_params", [{"model": "fireworks_ai/accounts/fireworks/models/gpt-oss-120b"}])`
`7`	`9`	`@evaluation_test(`
`8`	`10`	`input_messages=[`
`9`	`11`	`[`
`@@ -12,7 +14,6 @@`
`12`	`14`	`]`
`13`	`15`	`]`
`14`	`16`	`],`
`15`		`- completion_params=[{"model": "fireworks_ai/accounts/fireworks/models/gpt-oss-120b"}],`
`16`	`17`	`rollout_processor=SingleTurnRolloutProcessor(),`
`17`	`18`	`mode="all",`
`18`	`19`	`)`