chore: comments for view shapes

Co-authored-by: Sasha Abramowitz <reallysasha@gmail.com>
instadeepai · Oct 31, 2024 · 77f291c · 77f291c
1 parent 3d36aab
commit 77f291c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mava/wrappers/jumanji.py b/mava/wrappers/jumanji.py
@@ -419,6 +419,7 @@ def observation_spec(
             "step_count",
         )
 
+        # 2 sets of tiles in fov (blockers and targets) + xy position of agent and target
         tiles_in_fov = (self.fov * 2 + 1) ** 2
         single_agent_obs = 4 + tiles_in_fov * 2
         agents_view = specs.BoundedArray(