Yeah that makes sense thanks Violet

Author: hatkirby

worlds/lingo/__init__.py

@@ -89,17 +89,17 @@ def create_items(self):
                     pool.append(self.create_item("Puzzle Skip"))
 
             if traps:
-                trap_weights = {name: self.options.trap_weights.get(name, 0) for name in TRAP_ITEMS}
-                total_weight = sum(trap_weights.values())
+                total_weight = sum(self.options.trap_weights.values())
 
                 if total_weight == 0:
                     raise Exception("Sum of trap weights must be at least one.")
 
-                trap_counts = {name: int(weight * traps / total_weight) for name, weight in trap_weights.items()}
+                trap_counts = {name: int(weight * traps / total_weight)
+                               for name, weight in self.options.trap_weights.items()}
 
                 trap_difference = traps - sum(trap_counts.values())
                 if trap_difference > 0:
-                    allowed_traps = [name for name in TRAP_ITEMS if trap_weights[name] > 0]
+                    allowed_traps = [name for name in TRAP_ITEMS if self.options.trap_weights[name] > 0]
                     for i in range(0, trap_difference):
                         trap_counts[allowed_traps[i % len(allowed_traps)]] += 1