Skip to content

Commit 0e475f0

Browse files
vimuthVimuth Fernando
andauthored
In the random scheduler write the seed to file before running iteration if SHUTTLE_ALWAYS_PERSIST_SEED is set (#201)
Co-authored-by: Vimuth Fernando <vimuth@amazon.com>
1 parent a840dc4 commit 0e475f0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

shuttle/src/scheduler/random.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,9 @@ impl Scheduler for RandomScheduler {
8282
} else {
8383
self.iterations += 1;
8484
let seed = self.data_source.reinitialize();
85+
if let Ok(path) = std::env::var("SHUTTLE_ALWAYS_PERSIST_SEED") {
86+
std::fs::write(path, seed.to_string()).expect("Failed to write seed to file");
87+
}
8588
self.rng = Pcg64Mcg::seed_from_u64(seed);
8689
self.current_seed.update(seed);
8790
Some(Schedule::new(seed))

0 commit comments

Comments
 (0)