Fix bug in EnvParams.max_steps type (#39)

dunnolab · Nov 16, 2024 · a46e78c · a46e78c
1 parent c39870b
commit a46e78c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/xminigrid/environment.py b/src/xminigrid/environment.py
@@ -26,7 +26,7 @@ class EnvParams(struct.PyTreeNode):
     height: int = struct.field(pytree_node=False, default=9)
     width: int = struct.field(pytree_node=False, default=9)
     view_size: int = struct.field(pytree_node=False, default=7)
-    max_steps: Optional[None] = struct.field(pytree_node=False, default=None)
+    max_steps: Optional[int] = struct.field(pytree_node=False, default=None)
     render_mode: str = struct.field(pytree_node=False, default="rgb_array")