Skip to content

Commit e7a4270

Browse files
author
Ervin Teng
committed
Fix crash in trainer tests (trainer_metrics)
1 parent d41a2a0 commit e7a4270

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ml-agents/mlagents/trainers/tests/test_ppo.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,8 @@ def test_trainer_update_policy(mock_env, dummy_config, use_discrete):
366366
buffer.update_buffer["extrinsic_returns"] = buffer.update_buffer["rewards"]
367367
buffer.update_buffer["extrinsic_value_estimates"] = buffer.update_buffer["rewards"]
368368
trainer.training_buffer = buffer
369+
# Make sure trainer_metrics doesn't crash
370+
trainer.trainer_metrics.last_mean_return = 0
369371
trainer.update_policy()
370372
# Make batch length a larger multiple of sequence length
371373
trainer.trainer_parameters["batch_size"] = 128

0 commit comments

Comments
 (0)