From b8c30b05fe55345e9da32b481ad9c4a79aae4dfc Mon Sep 17 00:00:00 2001 From: Max Nigmatulin Date: Fri, 4 Aug 2023 17:50:37 +0300 Subject: [PATCH] Forgotten fix --- VSharp.ML.AIAgent/learning/play_game.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/VSharp.ML.AIAgent/learning/play_game.py b/VSharp.ML.AIAgent/learning/play_game.py index bc02df26f..47e1ca1e4 100644 --- a/VSharp.ML.AIAgent/learning/play_game.py +++ b/VSharp.ML.AIAgent/learning/play_game.py @@ -108,7 +108,7 @@ def play_map_with_stats( def play_game(with_predictor: Predictor, max_steps: int, maps_type: MapsType): with game_server_socket_manager() as ws: - maps = get_maps(websocket=ws, type=maps_type)[:4] + maps = get_maps(websocket=ws, type=maps_type) with tqdm.tqdm( total=len(maps), desc=f"{with_predictor.name():20}: {maps_type.value}",