Module moog_demos.example_configs.chase_avoid_torus

Chase/avoid task on a torus.

In this task there are predators (red circles), prey (yellow circles) and an agent (green square). The subject controls the agent with a joystick. The subject's goal is to catch the prey while avoiding being caught by the predators. The prey are repulsed by the agent and predators are attracted to the agent. Prey and predators move stochasticity and with constant speed.

The environment geometry is shaped like a torus — when an object reaches one boundary, it reappears on the opposite boundary.

Functions

def get_config(level)

Expand source code

def get_config(level):
    if level == 0:
        return _get_config(
            num_prey=1,
            num_predators=2,
        )
    elif level == 1:
        return _get_config(
            num_prey=lambda: np.random.randint(1, 3),
            num_predators=lambda: np.random.randint(1, 3),
        )
    else:
        raise ValueError('Invalid level {}'.format(level))