Skip to content
Snippets Groups Projects
Commit 2cb4c001 authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski
Browse files

Handle double > cases.

parent 038c9705
No related branches found
No related tags found
No related merge requests found
...@@ -114,6 +114,24 @@ def restore_collapse_edges(tree_tokens): ...@@ -114,6 +114,24 @@ def restore_collapse_edges(tree_tokens):
head, relation = d.split(':', 1) head, relation = d.split(':', 1)
ehead = f"{len(tree_tokens)}.{len(empty_tokens) + 1}" ehead = f"{len(tree_tokens)}.{len(empty_tokens) + 1}"
empty_node_relation, current_node_relation = relation.split(">", 1) empty_node_relation, current_node_relation = relation.split(">", 1)
# Edge case, double >
if ">" in current_node_relation:
second_empty_node_relation, current_node_relation = current_node_relation.split(">")
deps[i] = f"{ehead}:{current_node_relation}"
empty_tokens.append(
{
"id": ehead,
"deps": f"{head}:{empty_node_relation}"
}
)
empty_tokens.append(
{
"id": f"{len(tree_tokens)}.{len(empty_tokens) + 1}",
"deps": f"{ehead}:{second_empty_node_relation}"
}
)
else:
deps[i] = f"{ehead}:{current_node_relation}" deps[i] = f"{ehead}:{current_node_relation}"
empty_tokens.append( empty_tokens.append(
{ {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment