update env.py

This commit is contained in:
2025-11-25 10:04:13 +00:00
parent 9d4755a5fc
commit c6e9d27b4d

2
env.py
View File

@@ -1,4 +1,3 @@
```python
import re import re
import random import random
from typing import Any, Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple
@@ -249,4 +248,3 @@ class MazeConquerorsEnv(ta.Env):
return self.state.current_player_id, [] return self.state.current_player_id, []
def close(self) -> Tuple[Dict, Dict]: def close(self) -> Tuple[Dict, Dict]:
return self.state.rewards, self.state.game_info return self.state.rewards, self.state.game_info
```