MCPcopy Create free account
hub / github.com/google-deepmind/dm_control / after_step

Method after_step

dm_control/suite/base.py:79–83  ·  view source on GitHub ↗

Modifies colors according to the reward.

(self, physics)

Source from the content-addressed store, hash-verified

77 physics.set_control(action)
78
79 def after_step(self, physics):
80 """Modifies colors according to the reward."""
81 if self._visualize_reward:
82 reward = np.clip(self.get_reward(physics), 0.0, 1.0)
83 _set_reward_colors(physics, reward)
84
85 @property
86 def visualize_reward(self):

Callers 1

initialize_episodeMethod · 0.95

Calls 2

get_rewardMethod · 0.95
_set_reward_colorsFunction · 0.85

Tested by

no test coverage detected