Fixed return type
This commit is contained in:
parent
25372c2f34
commit
5d060ae8a4
@ -24,5 +24,5 @@ def pickWeightedAffinity(affinities: Affinities):
|
||||
population = [Classifiers.FOOD, Classifiers.WATER, Classifiers.REST]
|
||||
weights = affinities.getWeigths()
|
||||
|
||||
return random.choices(population, weights)
|
||||
return random.choices(population, weights)[0]
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user