fix: Channels dont change type from bool to int

strakam · Dec 4, 2024 · 0e540b6 · 0e540b6
1 parent e784a65
commit 0e540b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generals/core/channels.py b/generals/core/channels.py
@@ -37,7 +37,7 @@ def __init__(self, grid: np.ndarray, _agents: list[str]):
 
     def get_visibility(self, agent_id: str) -> np.ndarray:
         channel = self._ownership[agent_id]
-        return maximum_filter(channel, size=3)
+        return maximum_filter(channel, size=3).astype(bool)
 
     @staticmethod
     def channel_to_indices(channel: np.ndarray) -> np.ndarray: