sgoodfriend's picture
PPO playing LunarLander-v2 from https://github.com/sgoodfriend/rl-algo-impls/tree/0511de345b17175b7cf1ea706c3e05981f11761c
8310b64
raw
history blame contribute delete
No virus
678 Bytes
import gym
from rl_algo_impls.runner.config import Config
def import_for_env_id(env_id: str) -> None:
if "BulletEnv" in env_id:
import pybullet_envs
if "Microrts" in env_id:
import gym_microrts
def is_atari(config: Config) -> bool:
spec = gym.spec(config.env_id)
return "AtariEnv" in str(spec.entry_point)
def is_bullet_env(config: Config) -> bool:
return "BulletEnv" in config.env_id
def is_car_racing(config: Config) -> bool:
return "CarRacing" in config.env_id
def is_gym_procgen(config: Config) -> bool:
return "procgen" in config.env_id
def is_microrts(config: Config) -> bool:
return "Microrts" in config.env_id