Move generation of parses out of with blocks

This commit is contained in:
Ines Montani 2019-08-18 13:54:26 +02:00
parent ec0beccaf1
commit f35a8221d8
1 changed files with 4 additions and 4 deletions

View File

@ -84,12 +84,12 @@ def evaluate(
def render_parses(docs, output_path, model_name="", limit=250, deps=True, ents=True):
docs[0].user_data["title"] = model_name
if ents:
html = displacy.render(docs[:limit], style="ent", page=True)
with (output_path / "entities.html").open("w") as file_:
html = displacy.render(docs[:limit], style="ent", page=True)
file_.write(html)
if deps:
html = displacy.render(
docs[:limit], style="dep", page=True, options={"compact": True}
)
with (output_path / "parses.html").open("w") as file_:
html = displacy.render(
docs[:limit], style="dep", page=True, options={"compact": True}
)
file_.write(html)