fix(pre_commit): 🎨 auto format pre-commit hooks

roboflow · Sep 25, 2024 · 20f418f · 20f418f
1 parent 4629547
commit 20f418f
Showing 1 changed file with 2 additions and 14 deletions.
diff --git a/maestro/trainer/models/qwen2_vl/loaders.py b/maestro/trainer/models/qwen2_vl/loaders.py
@@ -1,4 +1,3 @@
-
 from transformers.pipelines.base import Dataset
 
 from maestro.trainer.common.data_loaders.datasets import JSONLDataset
@@ -17,18 +16,7 @@ def __getitem__(self, idx):
         suffix = data["suffix"]
         return {
             "messages": [
-                {
-                    "role": "user",
-                    "content": [
-                        {"type": "image", "image": image},
-                        {"type": "text", "text": prefix}
-                    ]
-                },
-                {
-                    "role": "assistant",
-                    "content": [
-                        {"type": "text", "text": suffix}
-                    ]
-                }
+                {"role": "user", "content": [{"type": "image", "image": image}, {"type": "text", "text": prefix}]},
+                {"role": "assistant", "content": [{"type": "text", "text": suffix}]},
             ]
         }