1566 - added correct num_train_episodes etc values to configs, fixed test_reward.py

2023-07-10 11:25:26 +01:00
parent af4fd4f66b
commit 9738d2b889
7 changed files with 24 additions and 24 deletions
--- a/tests/test_reward.py
+++ b/tests/test_reward.py
@@ -47,6 +47,6 @@ def test_rewards_are_being_penalised_at_each_step_function(
        Average Reward: -8 (-120 / 15)
    """
    with temp_primaite_session as session:
-        session.close()
+        session.evaluate()
        ev_rewards = session.eval_av_reward_per_episode_csv()
        assert ev_rewards[1] == -8.0