From c5f65acef34110605e8e9f6412d3f703772a42e5 Mon Sep 17 00:00:00 2001 From: evilchili Date: Sat, 27 Sep 2025 10:35:41 -0700 Subject: [PATCH] fix random --- random_sets/sets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/random_sets/sets.py b/random_sets/sets.py index 5c58da0..eadcaf6 100644 --- a/random_sets/sets.py +++ b/random_sets/sets.py @@ -24,11 +24,11 @@ class WeightedSet: def random(self) -> str: nonzero_members = [] nonzero_weights = [] - for i in range(self.weights): + for i in range(len(self.weights)): if float(self.weights[i]) == 0.0: continue - nozero_members.append(self.members[i]) - nozero_weights.append(self.weights[i]) + nonzero_members.append(self.members[i]) + nonzero_members.append(self.weights[i]) return random.choices(nonzero_members, nonzero_weights)[0] def __add__(self, obj):