🐛 Fix bug in displacy serve method (#46)

Signed-off-by: Marcos Martinez <Marcos.Martinez.Galindo@ibm.com>
IBM · Mar 30, 2023 · 311e1cf · 311e1cf
1 parent df6b2ca
commit 311e1cf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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('<span class="tex2jax_ignore">{}</span>'.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)