├── .gitignore ├── LICENSE ├── README.md ├── assets ├── Falcon.png └── Mcgill.png ├── docs ├── 2d_navigation_envs.md ├── _config.yml ├── benchmark_algos │ └── trpo.md ├── benchmarks.md ├── discrete_envs.md ├── falcon_rocket.md ├── index.md ├── mujoco_gym_envs.md └── wrappers.md ├── examples ├── EnvironmentGeneration.ipynb ├── image_based_navigation_example.py ├── manual_control_example.py ├── range_based_navigation_example.py └── state_based_navigation_example.py ├── gym_extensions ├── __init__.py ├── continuous │ ├── __init__.py │ ├── box2d │ │ ├── __init__.py │ │ └── falcon_drone_ship_lander.py │ ├── gym_navigation_2d │ │ ├── __init__.py │ │ ├── assets │ │ │ └── worlds_640x480_v0.pkl │ │ ├── env_generator.py │ │ ├── env_utils.py │ │ ├── geometry_utils.py │ │ ├── gym_rendering.py │ │ ├── image_based_navigation.py │ │ ├── range_based_navigation.py │ │ ├── rrt.py │ │ └── test_env_generator.py │ └── mujoco │ │ ├── __init__.py │ │ ├── gravity_envs.py │ │ ├── modified_ant.py │ │ ├── modified_arm.py │ │ ├── modified_half_cheetah.py │ │ ├── modified_hopper.py │ │ ├── modified_humanoid.py │ │ ├── modified_walker2d.py │ │ ├── perturbed_bodypart_env.py │ │ └── wall_envs.py ├── discrete │ ├── __init__.py │ └── classic │ │ ├── __init__.py │ │ ├── cartpole.py │ │ ├── cartpole_contextual.py │ │ └── pendulum_contextual.py ├── tests │ ├── README.md │ ├── __init__.py │ ├── test_contextual_envs.py │ ├── test_import.py │ ├── test_mujoco_import.py │ └── test_wrappers.py └── wrappers │ ├── __init__.py │ ├── normalized_env.py │ ├── observation_transform_wrapper.py │ ├── proxy_env.py │ ├── serializable.py │ └── transformers.py ├── multitask_benchmarks ├── __init__.py ├── commands_run ├── run_trpo.py └── utils.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/README.md -------------------------------------------------------------------------------- /assets/Falcon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/assets/Falcon.png -------------------------------------------------------------------------------- /assets/Mcgill.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/assets/Mcgill.png -------------------------------------------------------------------------------- /docs/2d_navigation_envs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/2d_navigation_envs.md -------------------------------------------------------------------------------- /docs/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/_config.yml -------------------------------------------------------------------------------- /docs/benchmark_algos/trpo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/benchmark_algos/trpo.md -------------------------------------------------------------------------------- /docs/benchmarks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/benchmarks.md -------------------------------------------------------------------------------- /docs/discrete_envs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/discrete_envs.md -------------------------------------------------------------------------------- /docs/falcon_rocket.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/falcon_rocket.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/mujoco_gym_envs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/mujoco_gym_envs.md -------------------------------------------------------------------------------- /docs/wrappers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/docs/wrappers.md -------------------------------------------------------------------------------- /examples/EnvironmentGeneration.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/examples/EnvironmentGeneration.ipynb -------------------------------------------------------------------------------- /examples/image_based_navigation_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/examples/image_based_navigation_example.py -------------------------------------------------------------------------------- /examples/manual_control_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/examples/manual_control_example.py -------------------------------------------------------------------------------- /examples/range_based_navigation_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/examples/range_based_navigation_example.py -------------------------------------------------------------------------------- /examples/state_based_navigation_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/examples/state_based_navigation_example.py -------------------------------------------------------------------------------- /gym_extensions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/__init__.py -------------------------------------------------------------------------------- /gym_extensions/continuous/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_extensions/continuous/box2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/box2d/__init__.py -------------------------------------------------------------------------------- /gym_extensions/continuous/box2d/falcon_drone_ship_lander.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/box2d/falcon_drone_ship_lander.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/__init__.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/assets/worlds_640x480_v0.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/assets/worlds_640x480_v0.pkl -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/env_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/env_generator.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/env_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/env_utils.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/geometry_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/geometry_utils.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/gym_rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/gym_rendering.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/image_based_navigation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/image_based_navigation.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/range_based_navigation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/range_based_navigation.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/rrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/rrt.py -------------------------------------------------------------------------------- /gym_extensions/continuous/gym_navigation_2d/test_env_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/gym_navigation_2d/test_env_generator.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/__init__.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/gravity_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/gravity_envs.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/modified_ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/modified_ant.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/modified_arm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/modified_arm.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/modified_half_cheetah.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/modified_half_cheetah.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/modified_hopper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/modified_hopper.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/modified_humanoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/modified_humanoid.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/modified_walker2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/modified_walker2d.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/perturbed_bodypart_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/perturbed_bodypart_env.py -------------------------------------------------------------------------------- /gym_extensions/continuous/mujoco/wall_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/continuous/mujoco/wall_envs.py -------------------------------------------------------------------------------- /gym_extensions/discrete/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_extensions/discrete/classic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/discrete/classic/__init__.py -------------------------------------------------------------------------------- /gym_extensions/discrete/classic/cartpole.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/discrete/classic/cartpole.py -------------------------------------------------------------------------------- /gym_extensions/discrete/classic/cartpole_contextual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/discrete/classic/cartpole_contextual.py -------------------------------------------------------------------------------- /gym_extensions/discrete/classic/pendulum_contextual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/discrete/classic/pendulum_contextual.py -------------------------------------------------------------------------------- /gym_extensions/tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/tests/README.md -------------------------------------------------------------------------------- /gym_extensions/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_extensions/tests/test_contextual_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/tests/test_contextual_envs.py -------------------------------------------------------------------------------- /gym_extensions/tests/test_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/tests/test_import.py -------------------------------------------------------------------------------- /gym_extensions/tests/test_mujoco_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/tests/test_mujoco_import.py -------------------------------------------------------------------------------- /gym_extensions/tests/test_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/tests/test_wrappers.py -------------------------------------------------------------------------------- /gym_extensions/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gym_extensions/wrappers/normalized_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/wrappers/normalized_env.py -------------------------------------------------------------------------------- /gym_extensions/wrappers/observation_transform_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/wrappers/observation_transform_wrapper.py -------------------------------------------------------------------------------- /gym_extensions/wrappers/proxy_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/wrappers/proxy_env.py -------------------------------------------------------------------------------- /gym_extensions/wrappers/serializable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/wrappers/serializable.py -------------------------------------------------------------------------------- /gym_extensions/wrappers/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/gym_extensions/wrappers/transformers.py -------------------------------------------------------------------------------- /multitask_benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multitask_benchmarks/commands_run: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/multitask_benchmarks/commands_run -------------------------------------------------------------------------------- /multitask_benchmarks/run_trpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/multitask_benchmarks/run_trpo.py -------------------------------------------------------------------------------- /multitask_benchmarks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/multitask_benchmarks/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Breakend/gym-extensions/HEAD/setup.py --------------------------------------------------------------------------------