chore: pre-commit

instadeepai · Oct 28, 2024 · bd4c8bc · bd4c8bc
1 parent f3c990e
commit bd4c8bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mava/networks/base.py b/mava/networks/base.py
@@ -271,7 +271,7 @@ def __call__(
         agent_qs: chex.Array,
         env_global_state: chex.Array,
     ) -> chex.Array:
-        B, T = agent_qs.shape[:2]  # batch size #noqa: N806
+        B, T = agent_qs.shape[:2]  # batch size
 
         agent_qs = jnp.reshape(agent_qs, (B, T, 1, self.num_agents))