chore: add back return typing from rebase

2025-05-14 10:39:46 +07:00
parent 67b1df21aa
commit 06fae0d34e
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/core/trainers/grpo/init.py
+++ b/src/axolotl/core/trainers/grpo/init.py
@@ -131,7 +131,7 @@ class GRPOStrategy:
        return None

    @classmethod
-    def get_blocklist_args_kwargs(cls):
+    def get_blocklist_args_kwargs(cls) -> list[str]:
        return ["dataset_num_proc", "max_length"]

    @classmethod