habitat.tasks.rearrange.rearrange_sensors.RearrangeReward class

An abstract class defining some measures that are always a part of any reward function in the Habitat 2.0 tasks.


def _get_count_coll_reward(self)
Count-based collision reward
def _want_count_coll(self)
Check if we want to consider penality from count-based collisions
def get_metric(self)
def reset_metric(self, *args, episode, task, observations, **kwargs)
def update_metric(self, *args, episode, task, observations, **kwargs)

Special methods

def __init__(self, *args, sim, config, task, **kwargs)


uuid: str = None

Method documentation

def habitat.tasks.rearrange.rearrange_sensors.RearrangeReward.get_metric(self)

Returns the current metric for Measure.