diff --git a/player.py b/player.py index 7603e7b..46bf6b3 100644 --- a/player.py +++ b/player.py @@ -81,7 +81,9 @@ class Player(object): def predictOpposingAction(self, opposingSkater): raise NotImplementedError() - def chooseAtkAction(self): + def chooseAtkAction(self, actionDic, opposingSkater): + """Picks an action/target node combo.""" + predAction = self.predictOpposingAction(opposingSkater) raise NotImplementedError() def chooseDefAction(self): diff --git a/team.py b/team.py index 9dbbfcd..3565bbc 100644 --- a/team.py +++ b/team.py @@ -3,8 +3,8 @@ from random import sample class Team(object): """A team of either 6 or 10 skaters and 1-3 goalies.""" - roster = [] #ordered, first line then second line; (#, name) - goalies = [] # (#, name) + roster:list[Player] = [] #ordered, first line then second line; (#, name) + goalies:Player = [] # (#, name) name = None shortname = None @@ -19,4 +19,4 @@ class Team(object): def chooseGoalie(self): - return sample(self.goalies,1) \ No newline at end of file + return sample(self.goalies,1)[0] \ No newline at end of file