Forgotten fix

PySymGym · Aug 4, 2023 · b8c30b0 · b8c30b0
1 parent 4e1defb
commit b8c30b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}",