admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 11:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

update losses names in tests

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-21 11:53:32 +02:00
parent a84b0e8132
commit a4c0c9e358
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/rl/test_actor_learner.py
+2 -2
View File
@@ -356,7 +356,7 @@ def test_learner_algorithm_wiring():
}
stats = algorithm.update(batch_iterator())
assert "critic" in stats.losses
assert "loss_critic" in stats.losses
# get_weights -> state_to_bytes round-trip
weights = algorithm.get_weights()
@@ -393,7 +393,7 @@ def test_learner_algorithm_wiring():
batch_size=batch_size,
)
trainer_stats = trainer.training_step()
assert "critic" in trainer_stats.losses
assert "loss_critic" in trainer_stats.losses
def test_initial_and_periodic_weight_push_consistency():