removed episodic_simple_reacher.py
This commit is contained in:
parent
e7dcdf38f1
commit
6169ede449
@ -1,4 +1,3 @@
|
|||||||
from alr_envs.classic_control.simple_reacher import SimpleReacherEnv
|
from alr_envs.classic_control.simple_reacher import SimpleReacherEnv
|
||||||
from alr_envs.classic_control.episodic_simple_reacher import EpisodicSimpleReacherEnv
|
|
||||||
from alr_envs.classic_control.viapoint_reacher import ViaPointReacher
|
from alr_envs.classic_control.viapoint_reacher import ViaPointReacher
|
||||||
from alr_envs.classic_control.hole_reacher import HoleReacherEnv
|
from alr_envs.classic_control.hole_reacher import HoleReacherEnv
|
||||||
|
@ -1,46 +0,0 @@
|
|||||||
from alr_envs.classic_control.simple_reacher import SimpleReacherEnv
|
|
||||||
from gym import spaces
|
|
||||||
import numpy as np
|
|
||||||
|
|
||||||
|
|
||||||
class EpisodicSimpleReacherEnv(SimpleReacherEnv):
|
|
||||||
def __init__(self, n_links, random_start=True):
|
|
||||||
super(EpisodicSimpleReacherEnv, self).__init__(n_links, random_start)
|
|
||||||
|
|
||||||
# self._goal_pos = None
|
|
||||||
|
|
||||||
if random_start:
|
|
||||||
state_bound = np.hstack([
|
|
||||||
[np.pi] * self.n_links, # cos
|
|
||||||
[np.pi] * self.n_links, # sin
|
|
||||||
[np.inf] * self.n_links, # velocity
|
|
||||||
])
|
|
||||||
else:
|
|
||||||
state_bound = np.empty(0, )
|
|
||||||
|
|
||||||
state_bound = np.hstack([
|
|
||||||
state_bound,
|
|
||||||
[np.inf] * 2, # x-y coordinates of goal
|
|
||||||
])
|
|
||||||
|
|
||||||
self.observation_space = spaces.Box(low=-state_bound, high=state_bound, shape=state_bound.shape)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def init_qpos(self):
|
|
||||||
return self._start_pos
|
|
||||||
|
|
||||||
# @property
|
|
||||||
# def goal_pos(self):
|
|
||||||
# return self._goal_pos
|
|
||||||
|
|
||||||
def _get_obs(self):
|
|
||||||
if self.random_start:
|
|
||||||
theta = self._joint_angles
|
|
||||||
return np.hstack([
|
|
||||||
np.cos(theta),
|
|
||||||
np.sin(theta),
|
|
||||||
self._angle_velocity,
|
|
||||||
self._goal,
|
|
||||||
])
|
|
||||||
else:
|
|
||||||
return self._goal
|
|
Loading…
Reference in New Issue
Block a user