From ffa783bff1b93292f73f33ab77d50768017ea135 Mon Sep 17 00:00:00 2001 From: Nolan Darilek Date: Tue, 3 Aug 2021 08:57:40 -0500 Subject: [PATCH] Tweak spawn table creation to not overflow. --- src/core.rs | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/core.rs b/src/core.rs index 58c1000..7aacf6a 100644 --- a/src/core.rs +++ b/src/core.rs @@ -522,9 +522,11 @@ impl RandomTable where T: Clone, { - pub fn add(&mut self, value: T, weight: u32) -> &mut Self { - for _ in 0..weight { - self.0.push(value.clone()); + pub fn add(&mut self, value: T, weight: i32) -> &mut Self { + if weight > 0 { + for _ in 0..weight { + self.0.push(value.clone()); + } } self }