Fixes data_type.gen

OffchainLabs · Jan 10, 2025 · 35ae6f3 · 35ae6f3
1 parent 34ef4c2
commit 35ae6f3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/arbitrator/tools/stylus_benchmark/src/scenarios/data_type.rs b/arbitrator/tools/stylus_benchmark/src/scenarios/data_type.rs
@@ -19,9 +19,8 @@ impl Rand for DataType {
     fn gen(&self) -> usize {
         let mut rng = rand::thread_rng();
         match self {
-            // makes sure that the generated number fits a signed integer
-            DataType::I32 => (rng.gen::<u32>() / 2 - 1).try_into().unwrap(),
-            DataType::I64 => (rng.gen::<u64>() / 2 - 1).try_into().unwrap(),
+            DataType::I32 => rng.gen_range(0..i32::MAX).try_into().unwrap(),
+            DataType::I64 => rng.gen_range(0..i64::MAX).try_into().unwrap(),
         }
     }
 }