formatting

WorldCereal · Nov 19, 2024 · fcfef83 · fcfef83
1 parent a53b5de
commit fcfef83
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_presto.py b/tests/test_presto.py
@@ -623,7 +623,9 @@ def forward(x, dynamic_world, mask):
             x, orig_indices, upd_mask = self.forward_encoder(
                 x, dynamic_world, mask, eval_task=False
             )
-            x = self.model.decoder.add_masked_tokens(x.to(device), orig_indices.to(device), upd_mask.to(device))
+            x = self.model.decoder.add_masked_tokens(
+                x.to(device), orig_indices.to(device), upd_mask.to(device)
+            )
             return self.model.decoder.reconstruct_inputs(x)
 
         batch_size, timesteps = 2, 3