Upload train_qwen3_telecom.py with huggingface_hub

train_qwen3_telecom.py

@@ -13,22 +13,24 @@ raw_dataset = load_dataset('netop/TeleQnA', split='test')
 
 def format_for_sft(example):
     """Convert TeleQnA format to chat messages format"""
-    # Build the question with options
-    options_text = []
-    for i in range(1, 6):  # Handle up to 5 options
-        option_key = f'option {i}'
-        if option_key in example and example[option_key]:
-            options_text.append(f'{i}. {example[option_key]}')
+    # Dataset columns: question, choices, answer, subject, explaination
+
+    # Build the question with choices
+    choices_text = []
+    if 'choices' in example and example['choices']:
+        for i, choice in enumerate(example['choices'], 1):
+            choices_text.append(f'{i}. {choice}')
 
     question_with_options = f"""{example['question']}
 
 Options:
-{chr(10).join(options_text)}"""
+{chr(10).join(choices_text)}"""
 
-    # Build the answer with explanation
+    # Build the answer with explanation (note: misspelled as 'explaination' in dataset)
+    explanation = example.get('explaination', '') or example.get('explanation', '')
     answer_text = f"""{example['answer']}
 
-Explanation: {example['explanation']}"""
+Explanation: {explanation}"""
 
     # Format as chat messages
     return {