Sanitize node labels in get_nps_from_treebank.py

kornai · Jul 31, 2018 · cb27847 · cb27847
1 parent 18440bf
commit cb27847
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/exp/alto/tools/get_nps_from_treebank.py b/exp/alto/tools/get_nps_from_treebank.py
@@ -5,6 +5,7 @@
 from common import sanitize_word
 
 def sanitize_tree(tree):
+    tree.set_label(sanitize_word(tree.label()))
     if tree.height() == 2: #word, pos
         tree[0] = sanitize_word(tree[0]) #tree[0] == word
     else: