diff --git a/gym_tool_use/transfers.py b/gym_tool_use/transfers.py index 3da44d6..874a66e 100644 --- a/gym_tool_use/transfers.py +++ b/gym_tool_use/transfers.py @@ -71,6 +71,7 @@ def structural_color_transfer(colors, np_random): """ # Force the colors to be random and unique. color_keys = trap_tube_env.SYMBOLIC_OBJECTS + [trap_tube_env.GROUND] + color_keys.pop(color_keys.index(trap_tube_env.TUBE2)) color_values = [] exclude_color_values = [ trap_tube_env.AGENT_COLOR, trap_tube_env.FOOD_COLOR] @@ -89,6 +90,7 @@ def structural_color_transfer(colors, np_random): color_values.append(value) new_colors = dict(zip(color_keys, color_values)) + new_colors[trap_tube_env.TUBE2] = new_colors[trap_tube_env.TUBE1] colors.update(new_colors) return colors diff --git a/gym_tool_use/trap_tube_env.py b/gym_tool_use/trap_tube_env.py index b263f71..9eecfea 100644 --- a/gym_tool_use/trap_tube_env.py +++ b/gym_tool_use/trap_tube_env.py @@ -503,7 +503,7 @@ class BaseTrapTubeEnv(gym_pycolab.PyColabEnv): def __init__(self, max_iterations=50, - delay=100, + delay=250, resize_scale=32, default_reward=0.): super(BaseTrapTubeEnv, self).__init__(