wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix indentation issue

This commit is contained in:
AdilZouitine 2025-04-03 16:05:29 +00:00
parent 38a8dbd9c9
commit e86fe66dbd
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lerobot/scripts/server/learner_server.py
View File

@ -381,11 +381,11 @@ def add_actor_information_and_train(
# Sample from the iterators # Sample from the iterators
batch = next(online_iterator) batch = next(online_iterator)
if dataset_repo_id is not None: if dataset_repo_id is not None:
batch_offline = next(offline_iterator) batch_offline = next(offline_iterator)
batch = concatenate_batch_transitions( batch = concatenate_batch_transitions(
left_batch_transitions=batch, right_batch_transition=batch_offline left_batch_transitions=batch, right_batch_transition=batch_offline
) )
actions = batch["action"] actions = batch["action"]
rewards = batch["reward"] rewards = batch["reward"]
@ -435,8 +435,8 @@ def add_actor_information_and_train(
# Update target networks # Update target networks
policy.update_target_networks() policy.update_target_networks()
# Sample for the last update in the UTD ratio # Sample for the last update in the UTD ratio
batch = next(online_iterator) batch = next(online_iterator)
if dataset_repo_id is not None: if dataset_repo_id is not None:
batch_offline = next(offline_iterator) batch_offline = next(offline_iterator)