Update make_dataset.py

src/data/make_dataset.py

@@ -11,6 +11,7 @@ def make_dataset(dataset="cnn_dailymail", split="train"):
         os.makedirs("data/raw")
     dataset = load_dataset(dataset, "3.0.0", split=split)
     df = pd.DataFrame()
+    df["highlights"] = dataset["highlights"]
     df.to_csv("data/raw/{}.csv".format(split))