Fixes data_type.gen

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