try xmlr

Riccorl · Oct 14, 2020 · 7fc2714 · 7fc2714
1 parent e61bcb6
commit 7fc2714
Show file tree

Hide file tree

Showing 2 changed files with 140 additions and 139 deletions.
diff --git a/transformer_srl/dataset_readers.py b/transformer_srl/dataset_readers.py
@@ -256,7 +256,6 @@ def _read(self, file_path: str):
                 "Filtering to only include file paths containing the %s domain",
                 self._domain_identifier,
             )
-
         for sentence in self._ontonotes_subset(
             ontonotes_reader, file_path, self._domain_identifier
         ):
@@ -270,7 +269,9 @@ def _read(self, file_path: str):
                         f for f, v in zip(sentence.predicate_lemmas, verb_indicator) if v == 1
                     ]
                     if not all(v == 0 for v in verb_indicator):
-                        yield self.text_to_instance(tokens, verb_indicator, frames, lemmas, tags, sentence_id)
+                        yield self.text_to_instance(
+                            tokens, verb_indicator, frames, lemmas, tags, sentence_id
+                        )
 
     def text_to_instance(  # type: ignore
         self,
@@ -279,7 +280,7 @@ def text_to_instance(  # type: ignore
         frames: List[str] = None,
         lemmas: List[str] = None,
         tags: List[str] = None,
-        sentence_id = None
+        sentence_id=None,
     ) -> Instance:
         """
         We take `pre-tokenized` input here, along with a verb label.  The verb label should be a
@@ -384,9 +385,7 @@ def _get_predicate_labels(self, sentence, verb_indicator):
         for i, v in enumerate(verb_indicator):
             if v == 1:
                 label = (
-                    "{}.{}".format(
-                        sentence.predicate_lemmas[i], sentence.predicate_framenet_ids[i]
-                    )
+                    "{}.{}".format(sentence.predicate_lemmas[i], sentence.predicate_framenet_ids[i])
                     if sentence.predicate_framenet_ids[i].isdigit()
                     else sentence.predicate_framenet_ids[i]
                 )