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 }