Update make_dataset.py

src/data/make_dataset.py

@@ -8,6 +8,7 @@ def make_dataset(dataset='cnn_dailymail', split='train'):
     df = pd.DataFrame()
     df['output_text'] = dataset['target']
     return df
+    df['article'] = dataset['article']
 
 if __name__ == '__main__':
     make_dataset(dataset='cnn_dailymail', split='train', version="3.0.0")