fix translations in push_to_argilla

rodekruis · Apr 5, 2024 · 6c389b8 · 6c389b8
1 parent 3ae9495
commit 6c389b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sml/load.py b/sml/load.py
@@ -252,7 +252,7 @@ def push_to_argilla(self, messages, dataset_name, tags=None):
             inputs = {'Original message': message.text}
             if message.translations:
                 for translation in message.translations:
-                    inputs[f"Translation ({next(iter(translation))})"] = next(iter(translation.values()))
+                    inputs[f"Translated message"] = translation['text']
             inputs['Message number'] = ix + 1
             inputs['Channel'] = message.group