#2639 Fix reset signature

2024-05-30 15:34:46 +01:00
parent 85f03570f7
commit a161a3a5ec
1 changed files with 1 additions and 1 deletions
--- a/src/primaite/session/environment.py
+++ b/src/primaite/session/environment.py
@@ -83,7 +83,7 @@ class PrimaiteGymEnv(gymnasium.Env):
        with open(path, "w") as 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."""
        _LOGGER.info(
            f"Resetting environment, episode {self.episode_counter}, "