From 06fae0d34e39823ad56d45fb082ba81693c4384e Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Wed, 14 May 2025 10:39:46 +0700 Subject: [PATCH] chore: add back return typing from rebase --- src/axolotl/core/trainers/grpo/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/core/trainers/grpo/__init__.py b/src/axolotl/core/trainers/grpo/__init__.py index 1b240c6b4..d8e6140ed 100644 --- 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