Fixed reference id matcher.

kedz · Aug 28, 2019 · 5c9239f · 5c9239f
1 parent 4660dc2
commit 5c9239f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnsum/data/summarization_dataset.py b/nnsum/data/summarization_dataset.py
@@ -47,7 +47,7 @@ def _collect_references(self, references_dir):
 
         ref_paths = defaultdict(list)
         for path in references_dir.glob("*"):
-            ref_id = path.stem.rsplit(".")[0]
+            ref_id = path.stem.rsplit(".", 1)[0]
             ref_paths[ref_id].append(path)
 
         all_ref_paths = []