Skip to content

Commit 3b0e9f8

Browse files
committed
fix(env creators): gripper config types
1 parent a7c0232 commit 3b0e9f8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/rcs/envs/creators.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def __call__( # type: ignore
9292
env: gym.Env = RobotEnv(robot, ControlMode.JOINTS if collision_guard is not None else control_mode)
9393

9494
env = FR3HW(env)
95-
if isinstance(gripper_cfg, rcs.sim.SimRobotConfig):
95+
if isinstance(gripper_cfg, rcs.hw.FHConfig):
9696
gripper = rcs.hw.FrankaHand(ip, gripper_cfg)
9797
env = GripperWrapper(env, gripper, binary=True)
9898
elif isinstance(gripper_cfg, rcs.hand.tilburg_hand.THConfig):
@@ -199,7 +199,7 @@ def __call__( # type: ignore
199199
camera_set = SimCameraSet(simulation, camera_set_cfg)
200200
env = CameraSetWrapper(env, camera_set, include_depth=True)
201201

202-
if isinstance(gripper_cfg, rcs.sim.SimRobotConfig):
202+
if gripper_cfg is not None and isinstance(gripper_cfg, rcs.sim.SimGripperConfig):
203203
gripper = sim.SimGripper(simulation, "0", gripper_cfg)
204204
env = GripperWrapper(env, gripper, binary=True)
205205
env = GripperWrapperSim(env, gripper)

0 commit comments

Comments
 (0)