From 8cfd64cd418f875b262e60f9c65bc62f5f9594c5 Mon Sep 17 00:00:00 2001 From: lilkm Date: Sun, 24 Nov 2024 23:14:46 +0100 Subject: [PATCH] fix: set default value of block_gripper for each env --- gym_lowcostrobot/envs/pick_place_cube_env.py | 2 +- gym_lowcostrobot/envs/stack_two_cubes_env.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/gym_lowcostrobot/envs/pick_place_cube_env.py b/gym_lowcostrobot/envs/pick_place_cube_env.py index 07eee54..f81db48 100644 --- a/gym_lowcostrobot/envs/pick_place_cube_env.py +++ b/gym_lowcostrobot/envs/pick_place_cube_env.py @@ -81,7 +81,7 @@ def __init__( observation_mode="image", action_mode="joint", reward_type="sparse", - block_gripper=True, + block_gripper=False, distance_threshold=0.05, cube_xy_range=0.3, target_xy_range=0.3, diff --git a/gym_lowcostrobot/envs/stack_two_cubes_env.py b/gym_lowcostrobot/envs/stack_two_cubes_env.py index 8385938..ffb492f 100644 --- a/gym_lowcostrobot/envs/stack_two_cubes_env.py +++ b/gym_lowcostrobot/envs/stack_two_cubes_env.py @@ -80,7 +80,7 @@ def __init__( observation_mode="image", action_mode="joint", reward_type="sparse", - block_gripper=True, + block_gripper=False, distance_threshold=0.05, cube_xy_range=0.3, n_substeps=20,