default output dir to documents dir

This commit is contained in:
Rémi Louf authored and Julien Chaumond committed 2019-12-09 20:37:55 -05:00
1 parent 693606a75c
commit 3a9a9f7861
2 files changed
+8 -5

No files matched your search

@@ -39,6 +39,8 @@ class SummarizationDataset(Dataset):
self.documents = []
story_filenames_list = os.listdir(path)
for story_filename in story_filenames_list:
if "summary" in story_filename:
continue
path_to_story = os.path.join(path, story_filename)
if not os.path.isfile(path_to_story):
continue