Formatting.

allenai · Mar 11, 2024 · 4f62eee · 4f62eee
1 parent 58603a0
commit 4f62eee
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/objathor/dataset/generate_holodeck_features.py b/objathor/dataset/generate_holodeck_features.py
@@ -93,9 +93,11 @@ def __getitem__(self, idx: int) -> Dict[str, Union[int, torch.Tensor]]:
         item = {
             "idx": idx,
             "uid": uid,
-            "text": ann["description"]
-            if ann.get("description") is not None
-            else ann["description_auto"],
+            "text": (
+                ann["description"]
+                if ann.get("description") is not None
+                else ann["description_auto"]
+            ),
         }
 
         if self.image_preprocessor is not None: