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

for more information, see https://pre-commit.ci

lerobot/common/envs/utils.py

@@ -108,9 +108,7 @@ def check_env_attributes_and_types(env: gym.vector.VectorEnv) -> None:
     with warnings.catch_warnings():
         warnings.simplefilter("once", UserWarning)  # Apply filter only in this function
 
-        if not (
-            hasattr(env.envs[0], "task_description") and hasattr(env.envs[0], "task")
-        ):
+        if not (hasattr(env.envs[0], "task_description") and hasattr(env.envs[0], "task")):
             warnings.warn(
                 "The environment does not have 'task_description' and 'task'. Some policies require these features.",
                 UserWarning,
@@ -124,9 +122,7 @@ def check_env_attributes_and_types(env: gym.vector.VectorEnv) -> None:
             )
 
 
-def add_envs_task(
-    env: gym.vector.VectorEnv, observation: dict[str, Any]
-) -> dict[str, Any]:
+def add_envs_task(env: gym.vector.VectorEnv, observation: dict[str, Any]) -> dict[str, Any]:
     """Adds task feature to the observation dict with respect to the first environment attribute."""
     if hasattr(env.envs[0], "task_description"):
         observation["task"] = env.call("task_description")

lerobot/common/policies/sac/modeling_sac.py

@@ -490,7 +490,7 @@ class SACObservationEncoder(nn.Module):
     """Encode image and/or state vector observations."""
 
     def __init__(self, config: SACConfig, input_normalizer: nn.Module) -> None:
-        super(SACObservationEncoder, self).__init__()
+        super().__init__()
         self.config = config
         self.input_normalization = input_normalizer
         self._init_image_layers()

lerobot/common/robot_devices/control_utils.py

@@ -24,8 +24,8 @@ from contextlib import nullcontext
 from copy import copy
 from functools import cache
 
-import rerun as rr
 import numpy as np
+import rerun as rr
 import torch
 from deepdiff import DeepDiff
 from termcolor import colored