Fix parse error in viterbi_tree.py

I installed treelearn but setup.py complained about a parse error in viterbi_tree.py.  Here's a bandaid to fix the parse error until the file is complete.