check if argument flat is true or not (#3156)

explosion · Jan 14, 2019 · 214c2ec · 214c2ec
1 parent d97661d
commit 214c2ec
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/spacy/tokens/printers.py b/spacy/tokens/printers.py
@@ -37,8 +37,9 @@ def POS_tree(root, light=False, flat=False):
  `merge_ents(doc)` ran on it.
  """
  subtree = format_POS(root, light=light, flat=flat)
- for c in root.children:
- subtree["modifiers"].append(POS_tree(c))
+ if not flat:
+ for c in root.children:
+ subtree["modifiers"].append(POS_tree(c))
  return subtree