From 300ffc2cb63bdbafe1d09c34014274b04474fa6d Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Mon, 3 Feb 2025 00:45:58 -0500
Subject: [PATCH] make it a dataclass

---
 src/axolotl/core/trainers/grpo/args.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/axolotl/core/trainers/grpo/args.py b/src/axolotl/core/trainers/grpo/args.py
index 3be21a4ce..e5f2cc254 100644
--- a/src/axolotl/core/trainers/grpo/args.py
+++ b/src/axolotl/core/trainers/grpo/args.py
@@ -1,12 +1,15 @@
 """
 Axolotl Specific Training Args
 """
+from dataclasses import dataclass
+
 from trl import GRPOConfig
 
 from axolotl.core.training_args import AxolotlTrainingMixins
 
 
-class AxolotlGRPOConfig(AxolotlTrainingMixins, GRPOConfig):
+@dataclass
+class AxolotlGRPOConfig(GRPOConfig, AxolotlTrainingMixins):
     """
     Axolotl GRPO Config for GRPO training
     """