import numpy as np SEED = 2026 def draw_batch(rng): return rng.integers(0, 20, size=8) rng = np.random.default_rng(seed=SEED) first_draw = draw_batch(rng) next_draw = draw_batch(rng) replay_rng = np.random.default_rng(seed=SEED) replayed_first_draw = draw_batch(replay_rng) print("seed:", SEED) print("first draw:", first_draw) print("next draw:", next_draw) print("replayed first draw:", replayed_first_draw) print("first draw matches replay:", np.array_equal(first_draw, replayed_first_draw)) print("next draw matches first:", np.array_equal(next_draw, first_draw))