From 87ac8bc31797ed4523f6a400402df2649fd0a7bd Mon Sep 17 00:00:00 2001 From: pzhokhov Date: Wed, 5 Sep 2018 14:03:13 -0700 Subject: [PATCH] install roboschool in install.py (#55) * putting instructions from README.md into a script * install roboschool as a part of setup.py * install roboschool from install.py * export pkg_config_path * remove compilation step from roboschool/setup.py * removed roboschool install from games install due to extra compilation step * removed unused import from roboschool/setup.py --- baselines/run.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/baselines/run.py b/baselines/run.py index 2210b51..faa2786 100644 --- a/baselines/run.py +++ b/baselines/run.py @@ -21,9 +21,19 @@ try: except ImportError: MPI = None +try: + import pybullet_envs +except ImportError: + pybullet_envs = None + +try: + import roboschool +except ImportError: + roboschool = None + _game_envs = defaultdict(set) for env in gym.envs.registry.all(): - # solve this with regexes + # TODO: solve this with regexes env_type = env._entry_point.split(':')[0].split('.')[-1] _game_envs[env_type].add(env.id) @@ -44,6 +54,7 @@ _game_envs['retro'] = { def train(args, extra_args): env_type, env_id = get_env_type(args.env) + print(f'env_type: {env_type}') total_timesteps = int(args.num_timesteps) seed = args.seed