Skip to content

Commit 6df0b5d

Browse files
authored
Fix algorithm configs for algorithms needing utility_function (#387)
1 parent cfd560b commit 6df0b5d

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

api/ai/interfaces/algorithm_config.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,19 +93,31 @@ def validate(self):
9393
class MultipleRoundRobinAlgorithmConfig(AlgorithmConfig):
9494
utility_function: Callable[[Student, TeamShell], float]
9595

96+
def __init__(self, utility_function: Callable[[Student, TeamShell], float]):
97+
super().__init__()
98+
self.utility_function = utility_function
99+
96100
def validate(self):
97101
super().validate()
98102

99103

100104
class DoubleRoundRobinAlgorithmConfig(AlgorithmConfig):
101105
utility_function: Callable[[Student, TeamShell], float]
102106

107+
def __init__(self, utility_function: Callable[[Student, TeamShell], float]):
108+
super().__init__()
109+
self.utility_function = utility_function
110+
103111
def validate(self):
104112
super().validate()
105113

106114

107115
class GeneralizedEnvyGraphAlgorithmConfig(AlgorithmConfig):
108116
utility_function: Callable[[Student, TeamShell], float]
109117

118+
def __init__(self, utility_function: Callable[[Student, TeamShell], float]):
119+
super().__init__()
120+
self.utility_function = utility_function
121+
110122
def validate(self):
111123
super().validate()

0 commit comments

Comments
 (0)