missing entity stuff

MSU-AI · Mar 11, 2024 · eadca1b · eadca1b
1 parent f88e1c0
commit eadca1b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/clash_royale/envs/game_engine/arena.py b/clash_royale/envs/game_engine/arena.py
@@ -8,7 +8,7 @@
 
 from typing import TYPE_CHECKING
 
-from clash_royale.envs.game_engine.entities.entity import EntityCollection
+from clash_royale.envs.game_engine.entities.entity import Entity, EntityCollection
 
 if TYPE_CHECKING:
     # Only import for typechecking to prevent circular dependency
@@ -43,3 +43,6 @@ def reset(self) -> None:
 
     def step(self) -> None:
         pass
+
+    def get_entities(self) -> List[Entity]:
+        pass
diff --git a/clash_royale/envs/game_engine/game_engine.py b/clash_royale/envs/game_engine/game_engine.py
@@ -66,7 +66,7 @@ def reset(self) -> None:
         Arena.reset() method missing.
         Player.reset() method missing.
         """
-        
+
         self.arena.reset()
         self.player1.reset(elixir=5)
         self.player2.reset(elixir=5)
@@ -84,7 +84,7 @@ def make_image(self, player_id: int) -> npt.NDArray[np.uint8]:
         arena.get_entities() missing
         """
 
-        entities = self.arena.get_entities()
+        entities: List[Entity] = self.arena.get_entities()
         canvas = pygame.Surface(size=self.resolution)
 
         #rendering logic goes here...