fix weightedsets random()
This commit is contained in:
parent
c5f65acef3
commit
b09b7907eb
|
@ -28,7 +28,7 @@ class WeightedSet:
|
|||
if float(self.weights[i]) == 0.0:
|
||||
continue
|
||||
nonzero_members.append(self.members[i])
|
||||
nonzero_members.append(self.weights[i])
|
||||
nonzero_weights.append(self.weights[i])
|
||||
return random.choices(nonzero_members, nonzero_weights)[0]
|
||||
|
||||
def __add__(self, obj):
|
||||
|
|
Loading…
Reference in New Issue
Block a user