oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#901 - merged dev into branch

Browse Source
This commit is contained in:
SunilSamra
2023-07-17 19:54:07 +01:00
parent 95b6211781 35af1e9d1e
commit a2f43b5abc
27 changed files with 1010 additions and 304 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_seeding_and_deterministic_session.py
View File

@@ -43,11 +43,9 @@ def test_seeded_learning(temp_primaite_session):
session._training_config.seed == 67890
), "Expected output is based upon a agent that was trained with seed 67890"
session.learn()
actual_mean_reward_per_episode = session.learn_av_reward_per_episode_dict()
print("\n")
print(session.learn_av_reward_per_episode())
assert expected_mean_reward_per_episode == session.learn_av_reward_per_episode()
assert actual_mean_reward_per_episode == expected_mean_reward_per_episode
@pytest.mark.skip(reason="Inconsistent results. Needs someone with RL knowledge to investigate further.")
@@ -62,5 +60,5 @@ def test_deterministic_evaluation(temp_primaite_session):
# do stuff
session.learn()
session.evaluate()
eval_mean_reward = session.eval_av_reward_per_episode_csv()
eval_mean_reward = session.eval_av_reward_per_episode_dict()
assert len(set(eval_mean_reward.values())) == 1