updated simple reacher example to new structure

This commit is contained in:
ottofabian 2021-06-24 15:26:20 +02:00
parent e1dc3eeddf
commit a30bdb8ce5

View File

@ -1,15 +1,15 @@
from typing import Iterable, Union from typing import Iterable, Union
import gym
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
from gym import spaces from gym import spaces
from gym.utils import seeding from gym.utils import seeding
from mp_env_api.envs.mp_env import MpEnv
from mp_env_api.envs.mp_env_wrapper import MPEnvWrapper from mp_env_api.envs.mp_env_wrapper import MPEnvWrapper
class SimpleReacherEnv(MpEnv): class SimpleReacherEnv(gym.Env):
""" """
Simple Reaching Task without any physics simulation. Simple Reaching Task without any physics simulation.
Returns no reward until 150 time steps. This allows the agent to explore the space, but requires precise actions Returns no reward until 150 time steps. This allows the agent to explore the space, but requires precise actions