evilchili/random-sets
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix random

Browse Source
This commit is contained in:
evilchili 2025-09-27 10:35:41 -07:00
parent 185375b51e
commit c5f65acef3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
random_sets/sets.py
View File

@ -24,11 +24,11 @@ class WeightedSet:
def random(self) -> str: def random(self) -> str:
nonzero_members = [] nonzero_members = []
nonzero_weights = [] nonzero_weights = []
for i in range(self.weights): for i in range(len(self.weights)):
if float(self.weights[i]) == 0.0: if float(self.weights[i]) == 0.0:
continue continue
nozero_members.append(self.members[i]) nonzero_members.append(self.members[i])
nozero_weights.append(self.weights[i]) nonzero_members.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):