edugp commited on
Commit
abd3459
·
1 Parent(s): b846822

Add random seed to shuffling of dataset

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ def hub_dataset_to_dataframe(path: str, name: str, split: str, text_column: str,
81
  load_dataset_fn = partial(load_dataset_fn, name=name)
82
  if split:
83
  load_dataset_fn = partial(load_dataset_fn, split=split)
84
- dataset = load_dataset_fn().shuffle()[:sample]
85
  return pd.DataFrame(dataset)
86
 
87
 
 
81
  load_dataset_fn = partial(load_dataset_fn, name=name)
82
  if split:
83
  load_dataset_fn = partial(load_dataset_fn, split=split)
84
+ dataset = load_dataset_fn().shuffle(seed=SEED)[:sample]
85
  return pd.DataFrame(dataset)
86
 
87