Fixed return type

This commit is contained in:
Marcin Kostrzewski 2020-05-16 10:54:31 +02:00
parent 25372c2f34
commit 5d060ae8a4

View File

@ -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]