Merge branch 'dev' into feature/2588-minimal-deps

2024-05-31 12:11:16 +01:00
parent 0c19546322 a5cd902175
commit c7b2ed7f86
3 changed files with 171 additions and 2 deletions
--- a/src/primaite/session/environment.py
+++ b/src/primaite/session/environment.py
@@ -82,7 +82,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}, "