diff --git a/giant/pipeline.py b/giant/pipeline.py index f24fcd4..b265dbe 100644 --- a/giant/pipeline.py +++ b/giant/pipeline.py @@ -70,7 +70,7 @@ def run_train_job( cache.event_index = (unique_ids, counts) train_events, val_events = make_event_split( - unique_ids, val_fraction=t["val_fraction"] + unique_ids, val_fraction=t["val_fraction"], seed=t["seed"] ) events_arr = np.array(sorted(train_events)) n_train_steps = setup_cache.n_train_steps_for_split(unique_ids, counts, events_arr)