From a8838c081bc2d244a76f0bfb04331eea47a8ee89 Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Mon, 13 Apr 2026 19:06:28 +0200 Subject: [PATCH] =?UTF-8?q?perf:=20remove=20redundant=20CPU=E2=86=92GPU?= =?UTF-8?q?=E2=86=92CPU=20transition=20move=20in=20learner?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/lerobot/rl/learner.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/src/lerobot/rl/learner.py b/src/lerobot/rl/learner.py index 39cd70b0b..6c8a0a95f 100644 --- a/src/lerobot/rl/learner.py +++ b/src/lerobot/rl/learner.py @@ -97,7 +97,6 @@ from lerobot.utils.train_utils import ( save_checkpoint, update_last_checkpoint, ) -from lerobot.utils.transition import move_transition_to_device from lerobot.utils.utils import ( format_big_number, init_logging, @@ -382,7 +381,6 @@ def add_actor_information_and_train( transition_queue=transition_queue, replay_buffer=replay_buffer, offline_replay_buffer=offline_replay_buffer, - device=device, dataset_repo_id=dataset_repo_id, shutdown_event=shutdown_event, ) @@ -906,7 +904,6 @@ def process_transitions( transition_queue: Queue, replay_buffer: ReplayBuffer, offline_replay_buffer: ReplayBuffer, - device: str, dataset_repo_id: str | None, shutdown_event: any, ): @@ -916,7 +913,6 @@ def process_transitions( transition_queue: Queue for receiving transitions from the actor replay_buffer: Replay buffer to add transitions to offline_replay_buffer: Offline replay buffer to add transitions to - device: Device to move transitions to dataset_repo_id: Repository ID for dataset shutdown_event: Event to signal shutdown """ @@ -925,8 +921,6 @@ def process_transitions( transition_list = bytes_to_transitions(buffer=transition_list) for transition in transition_list: - transition = move_transition_to_device(transition=transition, device=device) - # Skip transitions with NaN values if check_nan_in_transition( observations=transition["state"],