#1648 - Reverted benchmark sessions and episodes numbers

2023-07-21 15:07:21 +01:00
parent 4bb4850681
commit 1c6ceaa95b
2 changed files with 2 additions and 2 deletions
--- a/benchmark/primaite_benchmark.py
+++ b/benchmark/primaite_benchmark.py
@@ -414,7 +414,7 @@ def run():
    """Run the PrimAITE benchmark."""
    start_datetime = datetime.now()
    av_reward_per_episode_dicts = {}
-    for i in range(1, 3):
+    for i in range(1, 11):
        print(f"Starting Benchmark Session: {i}")
        with _get_benchmark_primaite_session() as session:
            session.learn()