Custom OpenAI gym environment
cd CustomGym
pip install -e .
Just install and import the package custom_gym
and here you go!
import gym
import custom_gym
from custom_gym import RecorderWrapper
# classic_control
env = gym.make('MassPointGoal-v0')
# recorder wrapper
env = RecorderWrapper(env, './test_data/', file_format='json')
See detail example in test.py
.
- OpenAI Gym
Actually this project is following the tutroial ofgym
.