sarsa update and flake8 error

automl-edu · Nov 9, 2023 · efdea61 · efdea61
1 parent 7d8d969
commit efdea61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rl_exercises/week_4/sarsa.py b/rl_exercises/week_4/sarsa.py
@@ -46,8 +46,8 @@ def __init__(self, env: gym.Env, policy: EpsilonGreedyPolicy, alpha: float = 0.5
 
     def predict_action(self, state: np.array, info: dict = {}, evaluate: bool = False) -> Any:  # type: ignore # noqa
         """Predict the action for a given state"""
+        action = info
         action = self.policy(self.Q, state, eval=evaluate)  # type: ignore
-        info = {}
         return action
 
     def save(self, path: str) -> Any:  # type: ignore
@@ -72,7 +72,7 @@ def load(self, path) -> Any:  # type: ignore
         """
         self.Q = np.load(path)
 
-    def update(  # type: ignore
+    def update_agent(  # type: ignore
         self,
         transition: list[np.array],  # type: ignore
         next_action: int,