Source code for soccer_strategy.obstacle

import enum

import numpy as np


[docs]class Obstacle:
[docs]    class Team(enum.IntEnum):
        UNKNOWN = 0
        FRIENDLY = 1
        OPPONENT = 2

[docs]    def __init__(self, position=np.array([0, 0]), team=Team.UNKNOWN, probability: float = 1):
        self.position = position
        self.team = team
        self.probability = probability