Add type annotations for metadata and render_mode (#3007)

This commit is contained in:
Alexander Zhang
2022-08-01 04:53:41 -07:00
committed by GitHub
parent 96e6cd66df
commit dce35dfa92

View File

@@ -107,8 +107,9 @@ class Env(Generic[ObsType, ActType], metaclass=decorator):
"""
# Set this in SOME subclasses
metadata = {"render_modes": []}
render_mode = None # define render_mode if your environment supports rendering
metadata: Dict[str, Any] = {"render_modes": []}
# define render_mode if your environment supports rendering
render_mode: Optional[str] = None
reward_range = (-float("inf"), float("inf"))
spec: "EnvSpec" = None