diff --git a/random_sets/sets.py b/random_sets/sets.py index eadcaf6..cd07d8b 100644 --- a/random_sets/sets.py +++ b/random_sets/sets.py @@ -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):