(self, physics)
Modifies colors according to the reward.
Modifies colors according to the reward.
Modifies colors according to the reward .
def after_step(self, physics):
    """Modifies colors according to the reward."""
    if self._visualize_reward:
      reward = np.clip(self.get_reward(physics), 0.0, 1.0)
      _set_reward_colors(physics, reward)
def after_step ( self , physics ) : if self . _visualize_reward : reward = np . clip ( self . get_reward ( physics ) , 0.0 , 1.0 ) _set_reward_colors ( physics , reward )