[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
2025-04-02 08:38:38 +00:00 · 2025-04-02 08:38:38 +00:00 · 0b68f2c2fb
parent eea7cc424a
commit 0b68f2c2fb
1 changed files with 4 additions and 1 deletions
--- a/lerobot/common/policies/pi0fast/modeling_pi0fast.py
+++ b/lerobot/common/policies/pi0fast/modeling_pi0fast.py
@ -846,7 +846,10 @@ class PI0FAST(nn.Module):
        decoded_actions = [
            torch.tensor(
                self.decode_actions_with_fast(
-                    tok.tolist(), time_horizon=action_horizon, action_dim=action_dim, relaxed_decoding=self.config.relaxed_decoding
+                    tok.tolist(),
+                    time_horizon=action_horizon,
+                    action_dim=action_dim,
+                    relaxed_decoding=self.config.relaxed_decoding,
                ),
                device=tokens.device,
            ).squeeze(0)