Update tests/e2e/solo/test_grpo.py

Co-authored-by: NanoCode012 <nano@axolotl.ai>
2025-04-08 09:07:04 -04:00
parent 0e1b081e49
commit 2798817cf9
1 changed files with 1 additions and 1 deletions
--- a/tests/e2e/solo/test_grpo.py
+++ b/tests/e2e/solo/test_grpo.py
@@ -44,7 +44,7 @@ def oai_gsm8k_transform(cfg, *args, **kwargs):
            )

    @with_temp_dir
-    def test_custom_rewards_fn_loading(self):
+    def test_custom_rewards_fn_preprocess(self, temp_dir):
        # pylint: disable=duplicate-code
        cfg = DictDefault(
            {