From f3750032bea2a5c600559fa4910e3031c41fdfdd Mon Sep 17 00:00:00 2001 From: SunilSamra Date: Tue, 11 Jul 2023 12:37:14 +0100 Subject: [PATCH] 1566 - applied pre-commit --- tests/test_train_eval_episode_steps.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_train_eval_episode_steps.py b/tests/test_train_eval_episode_steps.py index daa93055..b839e630 100644 --- a/tests/test_train_eval_episode_steps.py +++ b/tests/test_train_eval_episode_steps.py @@ -22,8 +22,8 @@ def test_eval_steps_differ_from_training(temp_primaite_session): num_eval_episodes = 1 """ expected_learning_metadata = {"total_episodes": 3, "total_time_steps": 75} - expected_evaluation_metadata = {"total_episodes": 1, "total_time_steps": 17} + with temp_primaite_session as session: # Run learning and check episode and step counts session.learn()