fix weightedsets random()

This commit is contained in:
evilchili 2025-09-27 10:42:56 -07:00
parent c5f65acef3
commit b09b7907eb

View File

@ -28,7 +28,7 @@ class WeightedSet:
if float(self.weights[i]) == 0.0: if float(self.weights[i]) == 0.0:
continue continue
nonzero_members.append(self.members[i]) 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] return random.choices(nonzero_members, nonzero_weights)[0]
def __add__(self, obj): def __add__(self, obj):