oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2639 Fix reset signature

Browse Source
This commit is contained in:
Marek Wolan
2024-05-30 15:34:46 +01:00
parent 85f03570f7
commit a161a3a5ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/session/environment.py
View File

@@ -83,7 +83,7 @@ class PrimaiteGymEnv(gymnasium.Env):
with open(path, "w") as file: with open(path, "w") as file:
json.dump(data, file) json.dump(data, file)
def reset(self, seed: Optional[int] = None) -> Tuple[ObsType, Dict[str, Any]]: def reset(self, seed: Optional[int] = None, options: Optional[Dict] = None) -> Tuple[ObsType, Dict[str, Any]]:
"""Reset the environment.""" """Reset the environment."""
_LOGGER.info( _LOGGER.info(
f"Resetting environment, episode {self.episode_counter}, " f"Resetting environment, episode {self.episode_counter}, "