From a161a3a5ec7f0782aa9c02752ce9373063410f57 Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Thu, 30 May 2024 15:34:46 +0100 Subject: [PATCH] #2639 Fix reset signature --- src/primaite/session/environment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/primaite/session/environment.py b/src/primaite/session/environment.py index 6c42c701..4d0544e9 100644 --- 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}, "