Expose the spacy language model for the word splitter in the se… (#3008)

allenai · Jun 26, 2019 · 7943b2f · 7943b2f
1 parent 9b929b2
commit 7943b2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/allennlp/predictors/semantic_role_labeler.py b/allennlp/predictors/semantic_role_labeler.py
@@ -15,9 +15,9 @@ class SemanticRoleLabelerPredictor(Predictor):
     """
     Predictor for the :class:`~allennlp.models.bidaf.SemanticRoleLabeler` model.
     """
-    def __init__(self, model: Model, dataset_reader: DatasetReader) -> None:
+    def __init__(self, model: Model, dataset_reader: DatasetReader, language: str = 'en_core_web_sm') -> None:
         super().__init__(model, dataset_reader)
-        self._tokenizer = SpacyWordSplitter(language='en_core_web_sm', pos_tags=True)
+        self._tokenizer = SpacyWordSplitter(language=language, pos_tags=True)
 
     def predict(self, sentence: str) -> JsonDict:
         """