Bladeren bron

type hint minigridenv init

Rodrigo Perez-Vicente 2 jaren geleden
bovenliggende
commit
7ef159eb62
1 gewijzigde bestanden met toevoegingen van 7 en 7 verwijderingen
  1. 7 7
      gym_minigrid/minigrid.py

+ 7 - 7
gym_minigrid/minigrid.py

@@ -653,13 +653,13 @@ class MiniGridEnv(gym.Env):
 
     def __init__(
         self,
-        grid_size=None,
-        width=None,
-        height=None,
-        max_steps=100,
-        see_through_walls=False,
-        agent_view_size=7,
-        render_mode=None,
+        grid_size: int=None,
+        width: int=None,
+        height: int=None,
+        max_steps: int=100,
+        see_through_walls: bool=False,
+        agent_view_size: int=7,
+        render_mode: str=None,
         **kwargs
     ):
         # Can't set both grid_size and width/height