Update pre-commit-config.yaml + pyproject.toml + ceil rerun & transformer dependencies version (#1520)

* chore: update .gitignore

* chore: update pre-commit

* chore(deps): update pyproject

* fix(ci): multiple fixes

* chore: pre-commit apply

* chore: address review comments

* Update pyproject.toml

Co-authored-by: Ben Zhang <5977478+ben-z@users.noreply.github.com>
Signed-off-by: Steven Palma <imstevenpmwork@ieee.org>

* chore(deps): add todo

---------

Signed-off-by: Steven Palma <imstevenpmwork@ieee.org>
Co-authored-by: Ben Zhang <5977478+ben-z@users.noreply.github.com>

src/lerobot/scripts/eval.py

@@ -50,12 +50,12 @@ import json
 import logging
 import threading
 import time
+from collections.abc import Callable
 from contextlib import nullcontext
 from copy import deepcopy
 from dataclasses import asdict
 from pathlib import Path
 from pprint import pformat
-from typing import Callable
 
 import einops
 import gymnasium as gym

src/lerobot/scripts/rl/crop_dataset_roi.py

@@ -18,7 +18,6 @@ import argparse
 import json
 from copy import deepcopy
 from pathlib import Path
-from typing import Dict, Tuple
 
 import cv2
 import torch
@@ -162,10 +161,10 @@ def get_image_from_lerobot_dataset(dataset: LeRobotDataset):
 
 def convert_lerobot_dataset_to_cropper_lerobot_dataset(
     original_dataset: LeRobotDataset,
-    crop_params_dict: Dict[str, Tuple[int, int, int, int]],
+    crop_params_dict: dict[str, tuple[int, int, int, int]],
     new_repo_id: str,
     new_dataset_root: str,
-    resize_size: Tuple[int, int] = (128, 128),
+    resize_size: tuple[int, int] = (128, 128),
     push_to_hub: bool = False,
     task: str = "",
 ) -> LeRobotDataset:

src/lerobot/scripts/rl/gym_manipulator.py

@@ -39,8 +39,9 @@ Example:
 import logging
 import time
 from collections import deque
+from collections.abc import Sequence
 from threading import Lock
-from typing import Annotated, Any, Sequence
+from typing import Annotated, Any
 
 import gymnasium as gym
 import numpy as np

src/lerobot/scripts/rl/learner.py

@@ -87,12 +87,10 @@ from lerobot.utils.process import ProcessSignalHandler
 from lerobot.utils.random_utils import set_seed
 from lerobot.utils.train_utils import (
     get_step_checkpoint_dir,
+    load_training_state as utils_load_training_state,
     save_checkpoint,
     update_last_checkpoint,
 )
-from lerobot.utils.train_utils import (
-    load_training_state as utils_load_training_state,
-)
 from lerobot.utils.transition import move_state_dict_to_device, move_transition_to_device
 from lerobot.utils.utils import (
     format_big_number,

src/lerobot/scripts/server/configs.py

@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from collections.abc import Callable
 from dataclasses import dataclass, field
-from typing import Callable
 
 import torch
 

src/lerobot/scripts/server/robot_client.py

@@ -36,10 +36,11 @@ import logging
 import pickle  # nosec
 import threading
 import time
+from collections.abc import Callable
 from dataclasses import asdict
 from pprint import pformat
 from queue import Queue
-from typing import Any, Callable, Optional
+from typing import Any
 
 import draccus
 import grpc
@@ -231,7 +232,7 @@ class RobotClient:
     def _aggregate_action_queues(
         self,
         incoming_actions: list[TimedAction],
-        aggregate_fn: Optional[Callable[[torch.Tensor, torch.Tensor], torch.Tensor]] = None,
+        aggregate_fn: Callable[[torch.Tensor, torch.Tensor], torch.Tensor] | None = None,
     ):
         """Finds the same timestep actions in the queue and aggregates them using the aggregate_fn"""
         if aggregate_fn is None:

src/lerobot/scripts/visualize_dataset.py

@@ -65,8 +65,8 @@ import argparse
 import gc
 import logging
 import time
+from collections.abc import Iterator
 from pathlib import Path
-from typing import Iterator
 
 import numpy as np
 import rerun as rr