From 164652697baf94f5e6a465a221f1212c0a8d4f65 Mon Sep 17 00:00:00 2001 From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com> Date: Wed, 13 Jan 2021 15:35:54 +0100 Subject: [PATCH] Fix dataset reader. --- combo/data/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/combo/data/dataset.py b/combo/data/dataset.py index 9aabe19..870659f 100644 --- a/combo/data/dataset.py +++ b/combo/data/dataset.py @@ -82,8 +82,8 @@ class UniversalDependenciesDatasetReader(allen_data.DatasetReader): for conllu_file in file_path: file = pathlib.Path(conllu_file) assert conllu_file and file.exists(), f"File with path '{conllu_file}' does not exists!" - with file.open("r"): - for annotation in conllu.parse_incr(file, fields=self.fields, field_parsers=self.field_parsers): + with file.open("r") as f: + for annotation in conllu.parse_incr(f, fields=self.fields, field_parsers=self.field_parsers): yield self.text_to_instance(annotation) @overrides -- GitLab