fix structural generation bug and slow down rendering.

fomorians · Jun 12, 2019 · e60825d · e60825d
1 parent e23e79c
commit e60825d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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__(