Update make_dataset.py

src/data/make_dataset.py

@@ -9,6 +9,7 @@ def make_dataset(dataset="cnn_dailymail", split="train"):
     """make dataset for summarisation"""
     if not os.path.exists("data/raw"):
         os.makedirs("data/raw")
+    dataset = load_dataset(dataset, "3.0.0", split=split)
     df = pd.DataFrame()
     df['article'] = dataset['article']
     df['highlights'] = dataset['highlights']