gym-notebook-wrapper 0.0.3

Wrapper for running/rendering OpenAI Gym on Jupyter Notebook

Gym-Notebook-Wrapper

Gym-Notebook-Wrapper provides small wrapper for running and rendering OpenAI Gym on Jupyter Notebook or similar (e.g. Google Corab).

Requirement

• Linux
• Xvfb
  • On Ubuntu, you can install sudo apt update && sudo apt install xvfb.
• Open GL (for some environment)
  • On Ubuntu, you can install sudo apt update && sudo apt install python-opengl

Installation

You can install from PyPI with pip install gym-notebook-wrapper

Usage

Wrap gym.Env class with gnwrapper.Animation. That's all! The render() method shows the environment on its output. An example code is following;

import gnwrapper
import gym

env = gnwrapper.Animation(gym.make('CartPole-v1'))

obs = env.reset()

for _ in range(1000):
    next_obs, reward, done, info = env.step(env.action_space.sample())
    env.render()

    obs = next_obs
    if done:
        obs = env.reset()

Limitation

• Calling render() method delete the other output for the same cell.
• The output image is shown only once.