chore: Mypy changes.

ch33nchan · Nov 22, 2021 · 784d437 · 784d437
1 parent 89ace75
commit 784d437
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mava/environment_loop.py b/mava/environment_loop.py
@@ -147,9 +147,9 @@ def _perform_turn(self, timestep: dm_env.TimeStep) -> dm_env.TimeStep:
         # save action, timestep pairs for current agent
         timestep = self._set_step_type(timestep, self._step_type[agent])
         self._agent_action_timestep[agent] = (
-            self._prev_action[agent],
+            self._prev_action[agent],  # type: ignore
             timestep,
-        )  # type: ignore
+        )
 
         self._prev_timestep[agent] = timestep
 
@@ -181,9 +181,9 @@ def _collect_last_timesteps(self, timestep: dm_env.TimeStep) -> None:
             agent = self._environment.current_agent
             timestep = self._set_step_type(timestep, dm_env.StepType.LAST)
             self._agent_action_timestep[agent] = (
-                self._prev_action[agent],
+                self._prev_action[agent],  # type: ignore
                 timestep,
-            )  # type: ignore
+            )
 
             timestep = self._environment.step(
                 generate_zeros_from_spec(self._environment.action_spec()[agent])