Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
from .BinArena import BinArena
from robosuite.models.objects import BreadObject, MilkObject, LemonObject, CanObject, BottleObject, CerealObject
from robosuite.models.robots import Sawyer
from robosuite.models.tasks import TableTopTask, UniformRandomSampler
from robosuite.wrappers import IKWrapper
except ImportError:
print("Robosuite is required to use this module. Please try ' pip install robosuite' ")
raise ImportError()
def make_sawyer_env(hparams_dict):
return IKWrapper(SawyerMultiObjEnv(**hparams_dict))
OBJECTS = [BreadObject, MilkObject, LemonObject, CanObject, BottleObject, CerealObject]
class SawyerMultiObjEnv(SawyerEnv):
"""
This class corresponds to the lifting task for the Sawyer robot arm.
"""
def __init__(
self,
gripper_type="TwoFingerGripper",
table_full_size=(0.8, 0.8, 0.8),
table_friction=(1., 5e-3, 1e-4),
use_camera_obs=True,
use_object_obs=True,
reward_shaping=False,
placement_initializer=None,