change _bits

mlcommons · Feb 7, 2024 · 1fc4ce2 · 1fc4ce2
1 parent 43b27b2
commit 1fc4ce2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/algorithmic_efficiency/random_utils.py b/algorithmic_efficiency/random_utils.py
@@ -69,9 +69,7 @@ def _check_jax_install() -> None:
 
 
 def _bits(seed: SeedType) -> int:
-  rng = np.random.RandomState(_signed_to_unsigned(seed))
-  b = rng.bytes(4)
-  return int.from_bytes(b, byteorder='little')
+  return seed
 
 
 def fold_in(seed: SeedType, data: int) -> SeedType: