From 4e53564670df5e590ae0b4bc9261e4f83433a33f Mon Sep 17 00:00:00 2001 From: SunilSamra Date: Fri, 14 Jul 2023 14:26:10 +0100 Subject: [PATCH] 901 - Changed the default expected_mean_reward_per_episode values in test_seeding_and_deterministic_session.py --- .../test_seeding_and_deterministic_session.py | 23 +++++++++++-------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/tests/test_seeding_and_deterministic_session.py b/tests/test_seeding_and_deterministic_session.py index 44ae2492..685e4c3e 100644 --- a/tests/test_seeding_and_deterministic_session.py +++ b/tests/test_seeding_and_deterministic_session.py @@ -26,16 +26,16 @@ def test_seeded_learning(temp_primaite_session): now work. If not, then you've got a bug :). """ expected_mean_reward_per_episode = { - 1: -33.90625, - 2: -32.32421875, - 3: -25.234375, - 4: -30.15625, - 5: -27.1484375, - 6: -29.609375, - 7: -29.921875, - 8: -29.3359375, - 9: -28.046875, - 10: -27.24609375, + 1: -30.703125, + 2: -29.94140625, + 3: -27.91015625, + 4: -29.66796875, + 5: -32.44140625, + 6: -30.33203125, + 7: -26.25, + 8: -22.44140625, + 9: -30.3125, + 10: -28.359375, } with temp_primaite_session as session: @@ -44,6 +44,9 @@ def test_seeded_learning(temp_primaite_session): ), "Expected output is based upon a agent that was trained with seed 67890" session.learn() + print("\n") + print(session.learn_av_reward_per_episode()) + assert expected_mean_reward_per_episode == session.learn_av_reward_per_episode()