gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-30 01:50:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix hand serialization tests (#1647)

Browse Source 
* fix serialization by moving EzPickle inheritance to the leaf classes and passing correct arguments to EzPickle

* fix touch sensor serialization too
This commit is contained in:
pzhokhov
2019-08-14 10:28:27 -07:00
committed by GitHub
parent f99ce5f324
commit cedecb35e3
2 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
gym/envs/robotics/hand/manipulate.py
View File

@@ -25,10 +25,10 @@ MANIPULATE_EGG_XML = os.path.join('hand', 'manipulate_egg.xml')
MANIPULATE_PEN_XML = os.path.join('hand', 'manipulate_pen.xml')
class ManipulateEnv(hand_env.HandEnv, utils.EzPickle):
class ManipulateEnv(hand_env.HandEnv):
def __init__(
self, model_path, target_position, target_rotation,
target_position_range, reward_type, initial_qpos={},
target_position_range, reward_type, initial_qpos=None,
randomize_initial_position=True, randomize_initial_rotation=True,
distance_threshold=0.01, rotation_threshold=0.1, n_substeps=20, relative_control=False,
ignore_z_target_rotation=False,
@@ -71,11 +71,12 @@ class ManipulateEnv(hand_env.HandEnv, utils.EzPickle):
assert self.target_position in ['ignore', 'fixed', 'random']
assert self.target_rotation in ['ignore', 'fixed', 'xyz', 'z', 'parallel']
initial_qpos = initial_qpos or {}
hand_env.HandEnv.__init__(
self, model_path, n_substeps=n_substeps, initial_qpos=initial_qpos,
relative_control=relative_control)
utils.EzPickle.__init__(self)
def _get_achieved_goal(self):
# Object position and rotation.
@@ -271,27 +272,30 @@ class ManipulateEnv(hand_env.HandEnv, utils.EzPickle):
}
class HandBlockEnv(ManipulateEnv):
class HandBlockEnv(ManipulateEnv, utils.EzPickle):
def __init__(self, target_position='random', target_rotation='xyz', reward_type='sparse'):
super(HandBlockEnv, self).__init__(
utils.EzPickle.__init__(self, target_position, target_rotation, reward_type)
ManipulateEnv.__init__(self,
model_path=MANIPULATE_BLOCK_XML, target_position=target_position,
target_rotation=target_rotation,
target_position_range=np.array([(-0.04, 0.04), (-0.06, 0.02), (0.0, 0.06)]),
reward_type=reward_type)
class HandEggEnv(ManipulateEnv):
class HandEggEnv(ManipulateEnv, utils.EzPickle):
def __init__(self, target_position='random', target_rotation='xyz', reward_type='sparse'):
super(HandEggEnv, self).__init__(
utils.EzPickle.__init__(self, target_position, target_rotation, reward_type)
ManipulateEnv.__init__(self,
model_path=MANIPULATE_EGG_XML, target_position=target_position,
target_rotation=target_rotation,
target_position_range=np.array([(-0.04, 0.04), (-0.06, 0.02), (0.0, 0.06)]),
reward_type=reward_type)
class HandPenEnv(ManipulateEnv):
class HandPenEnv(ManipulateEnv, utils.EzPickle):
def __init__(self, target_position='random', target_rotation='xyz', reward_type='sparse'):
super(HandPenEnv, self).__init__(
utils.EzPickle.__init__(self, target_position, target_rotation, reward_type)
ManipulateEnv.__init__(self,
model_path=MANIPULATE_PEN_XML, target_position=target_position,
target_rotation=target_rotation,
target_position_range=np.array([(-0.04, 0.04), (-0.06, 0.02), (0.0, 0.06)]),