Gagan Bhatia commited on
Commit
c5a3a4d
1 Parent(s): 5466abc

Update make_dataset.py

Browse files
Files changed (1) hide show
  1. src/data/make_dataset.py +1 -0
src/data/make_dataset.py CHANGED
@@ -1,6 +1,7 @@
1
  import yaml
2
  from datasets import load_dataset
3
  import pandas as pd
 
4
 
5
 
6
  def make_dataset(dataset='cnn_dailymail', split='train'):
 
1
  import yaml
2
  from datasets import load_dataset
3
  import pandas as pd
4
+ import os
5
 
6
 
7
  def make_dataset(dataset='cnn_dailymail', split='train'):