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: