From 3f3d08e5a8e59f6388b521c7c0fabf99a5000644 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Thu, 12 Feb 2026 19:37:57 +0100 Subject: [PATCH] chore(style): fix pre-commit --- src/lerobot/policies/pi0/modeling_pi0.py | 3 +-- src/lerobot/policies/pi05/modeling_pi05.py | 3 +-- src/lerobot/policies/wall_x/qwen_model/qwen2_5_vl_moe.py | 1 + 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/lerobot/policies/pi0/modeling_pi0.py b/src/lerobot/policies/pi0/modeling_pi0.py index 450fe25ce..a8ae83c95 100644 --- a/src/lerobot/policies/pi0/modeling_pi0.py +++ b/src/lerobot/policies/pi0/modeling_pi0.py @@ -60,8 +60,7 @@ class ActionSelectKwargs(TypedDict, total=False): def _gated_residual(residual: torch.Tensor, hidden_states: torch.Tensor, gate: torch.Tensor) -> torch.Tensor: - """Gated residual connection: residual + gate * hidden_states. - """ + """Gated residual connection: residual + gate * hidden_states.""" return residual + gate.unsqueeze(-1) * hidden_states diff --git a/src/lerobot/policies/pi05/modeling_pi05.py b/src/lerobot/policies/pi05/modeling_pi05.py index 2b0965e39..d5be17079 100644 --- a/src/lerobot/policies/pi05/modeling_pi05.py +++ b/src/lerobot/policies/pi05/modeling_pi05.py @@ -59,8 +59,7 @@ class ActionSelectKwargs(TypedDict, total=False): def _gated_residual(residual: torch.Tensor, hidden_states: torch.Tensor, gate: torch.Tensor) -> torch.Tensor: - """Gated residual connection: residual + gate * hidden_states. - """ + """Gated residual connection: residual + gate * hidden_states.""" return residual + gate.unsqueeze(-1) * hidden_states diff --git a/src/lerobot/policies/wall_x/qwen_model/qwen2_5_vl_moe.py b/src/lerobot/policies/wall_x/qwen_model/qwen2_5_vl_moe.py index d0091a83d..a1309ea9a 100644 --- a/src/lerobot/policies/wall_x/qwen_model/qwen2_5_vl_moe.py +++ b/src/lerobot/policies/wall_x/qwen_model/qwen2_5_vl_moe.py @@ -30,6 +30,7 @@ from transformers.utils import ( from .configuration_qwen2_5_vl import Qwen2_5_VLConfig, Qwen2_5_VLVisionConfig + # TODO(Steven): SlidingWindowCache was removed in transformers v5. Define a placeholder so isinstance checks # always return False (which is the correct behavior when no sliding window cache is in use). class _SlidingWindowCachePlaceholder: