!249 update dataset processing related

Merge pull request !249 from 金勇旭/master
2025-06-27 07:27:31 +00:00
parent 5b8d1d4bf2
commit b8186e58a8
1 changed files with 1 additions and 1 deletions
--- a/src/openmind/flow/datasets/preprocess/generic.py
+++ b/src/openmind/flow/datasets/preprocess/generic.py
@ -112,7 +112,7 @@ def convert_alpaca(examples, datasets_attr: InstructionDatasetAttr, convert_syst
        if examples[datasets_attr.prompt][i]:
            content.append(examples[datasets_attr.prompt][i])

-        if examples[datasets_attr.query][i]:
+        if examples.get(datasets_attr.query, None) and examples[datasets_attr.query][i]:
            content.append(examples[datasets_attr.query][i])

        prompt.append({"role": "user", "content": "\n".join(content)})