From d3a8c2c24787f89b3dafa4f98612c1c375371570 Mon Sep 17 00:00:00 2001 From: AdilZouitine Date: Thu, 3 Apr 2025 16:05:29 +0000 Subject: [PATCH] fix indentation issue --- lerobot/scripts/server/learner_server.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/lerobot/scripts/server/learner_server.py b/lerobot/scripts/server/learner_server.py index 08baa6ea..65b1d9b8 100644 --- a/lerobot/scripts/server/learner_server.py +++ b/lerobot/scripts/server/learner_server.py @@ -381,11 +381,11 @@ def add_actor_information_and_train( # Sample from the iterators batch = next(online_iterator) - if dataset_repo_id is not None: - batch_offline = next(offline_iterator) - batch = concatenate_batch_transitions( - left_batch_transitions=batch, right_batch_transition=batch_offline - ) + if dataset_repo_id is not None: + batch_offline = next(offline_iterator) + batch = concatenate_batch_transitions( + left_batch_transitions=batch, right_batch_transition=batch_offline + ) actions = batch["action"] rewards = batch["reward"] @@ -435,8 +435,8 @@ def add_actor_information_and_train( # Update target networks policy.update_target_networks() - # Sample for the last update in the UTD ratio - batch = next(online_iterator) + # Sample for the last update in the UTD ratio + batch = next(online_iterator) if dataset_repo_id is not None: batch_offline = next(offline_iterator)