diff --git a/src/instructlab/sdg/generate_data.py b/src/instructlab/sdg/generate_data.py index bfb11b07..70a2e49d 100644 --- a/src/instructlab/sdg/generate_data.py +++ b/src/instructlab/sdg/generate_data.py @@ -51,7 +51,7 @@ def _get_question(logger, synth_example): if "question" in synth_example: return synth_example["question"] - if "output" not in synth_example: + if not synth_example.get("output"): raise utils.GenerateException( f"Error: output not found in synth_example: {synth_example}" )