From 00ebf2faf9b2efeac673c02cfeb265af36150150 Mon Sep 17 00:00:00 2001 From: Sunny Liu Date: Thu, 27 Mar 2025 13:29:17 -0400 Subject: [PATCH] message key checking --- src/axolotl/processing_strategies.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/axolotl/processing_strategies.py b/src/axolotl/processing_strategies.py index e91fe3968..041f60d71 100644 --- a/src/axolotl/processing_strategies.py +++ b/src/axolotl/processing_strategies.py @@ -150,9 +150,13 @@ class ProcessingStrategy: processed_examples = [] for example in examples: - if not ("messages" in example or "conversations" in example): + if not ( + "messages" in example + or "conversations" in example + or "question" in example + ): raise ValueError( - "Only `messages` and `conversations` message keys are currently supported." + "Only `messages`, `conversations`, and `question` message keys are currently supported." ) processed_example = None