Apply suggestion from @qgallouedec

Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>

Author: abidlabs

examples/trl-integration.py

@@ -14,20 +14,20 @@
 project_name = f"trackio-trl-demo-{suffix}"
 
 prompts = [
-    {"role": "user", "content": "What is the capital of France?"},
-    {"role": "user", "content": "Who wrote Hamlet?"},
-    {"role": "user", "content": "What is 2 + 2?"},
-    {"role": "user", "content": "What color is the sky?"},
-    {"role": "user", "content": "Name a primary color."},
-    {"role": "user", "content": "What is the largest planet?"},
+    [{"role": "user", "content": "What is the capital of France?"}],
+    [{"role": "user", "content": "Who wrote Hamlet?"}],
+    [{"role": "user", "content": "What is 2 + 2?"}],
+    [{"role": "user", "content": "What color is the sky?"}],
+    [{"role": "user", "content": "Name a primary color."}],
+    [{"role": "user", "content": "What is the largest planet?"}],
 ] * 4
 completions = [
-    {"role": "assistant", "content": "Paris."},
-    {"role": "assistant", "content": "Shakespeare."},
-    {"role": "assistant", "content": "4."},
-    {"role": "assistant", "content": "Blue."},
-    {"role": "assistant", "content": "Red."},
-    {"role": "assistant", "content": "Jupiter."},
+    [{"role": "assistant", "content": "Paris."}],
+    [{"role": "assistant", "content": "Shakespeare."}],
+    [{"role": "assistant", "content": "4."}],
+    [{"role": "assistant", "content": "Blue."}],
+    [{"role": "assistant", "content": "Red."}],
+    [{"role": "assistant", "content": "Jupiter."}],
 ] * 4
 dataset = Dataset.from_dict({"prompt": prompts, "completion": completions})