diff --git a/src/instructlab/eval/mt_bench_branch_generator.py b/src/instructlab/eval/mt_bench_branch_generator.py index 1a4a0fa..278a851 100644 --- a/src/instructlab/eval/mt_bench_branch_generator.py +++ b/src/instructlab/eval/mt_bench_branch_generator.py @@ -56,8 +56,9 @@ def generate(judge_model_name, branch, taxonomy_dir, output_dir): print(f"failed to load {qna_file}. skipping...") continue for ex in examples: - q, a = ex["question"], ex["answer"] + q, a = ex.get("question"), ex.get("answer") if q is None or a is None: + logger.warning("Skipping malformed file %s", qna_file) continue c = ex.get("context")