From e3b07402a7f191bbbbdbd63aed29003d5cb91c3f Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sat, 19 Aug 2023 21:20:10 -0400
Subject: [PATCH] make sure to define all the explicit positional args

---
 src/axolotl/utils/callbacks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/axolotl/utils/callbacks.py b/src/axolotl/utils/callbacks.py
index 3fb90f1e1..6bb42d5aa 100644
--- a/src/axolotl/utils/callbacks.py
+++ b/src/axolotl/utils/callbacks.py
@@ -152,6 +152,8 @@ def mmlu_eval_callback_factory(trainer, tokenizer):
         def on_evaluate(
             self,
             args: AxolotlTrainingArguments,
+            state: TrainerState,  # pylint: disable=unused-argument
+            control: TrainerControl,  # pylint: disable=unused-argument
             **kwargs,  # pylint: disable=unused-argument
         ):
             data_loader = trainer.get_eval_dataloader(mmlu_dataset)