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]
|
population = [Classifiers.FOOD, Classifiers.WATER, Classifiers.REST]
|
||||||
weights = affinities.getWeigths()
|
weights = affinities.getWeigths()
|
||||||
|
|
||||||
return random.choices(population, weights)
|
return random.choices(population, weights)[0]
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user