Skip to content

timeout

Timeout

Bases: FailureCondition

Timeout (failure condition) Episode terminates if max_step steps have passed

Parameters:

Name Type Description Default
max_steps int

Maximum number of episode steps before timeout occurs

 500
Source code in omnigibson/termination_conditions/timeout.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Timeout(FailureCondition):
    """
    Timeout (failure condition)
    Episode terminates if max_step steps have passed

    Args:
        max_steps (int): Maximum number of episode steps before timeout occurs
    """

    def __init__(self, max_steps=500):
        # Store internal vars
        self._max_steps = max_steps

        # Run super
        super().__init__()

    def _step(self, task, env, action):
        # Terminate if number of steps passed exceeds threshold
        return env.episode_steps >= self._max_steps