Gagan Bhatia commited on
Commit
7654c8a
1 Parent(s): 7326d38

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
@@ -16,6 +16,7 @@ def make_dataset(dataset="cnn_dailymail", split="train"):
16
  df.to_csv("data/raw/{}.csv".format(split))
17
 
18
 
 
19
  with open("params.yml") as f:
20
  params = yaml.safe_load(f)
21
  pprint.pprint(params)
 
16
  df.to_csv("data/raw/{}.csv".format(split))
17
 
18
 
19
+ if __name__ == "__main__":
20
  with open("params.yml") as f:
21
  params = yaml.safe_load(f)
22
  pprint.pprint(params)