diff --git a/zshot/utils/displacy/displacy.py b/zshot/utils/displacy/displacy.py index ced7a40..367aa33 100644 --- a/zshot/utils/displacy/displacy.py +++ b/zshot/utils/displacy/displacy.py @@ -72,5 +72,8 @@ def _call_displacy(docs: Union[Iterable[Union[Doc]], Doc], style: str, method: s return display(HTML('{}'.format(html))) return html + if method == "render": + kwargs.update({"jupyter": jupyter}) + disp = getattr(s_displacy, method) - return disp(docs, style=style, options=options, jupyter=jupyter, **kwargs) + return disp(docs, style=style, options=options, **kwargs)