diff --git a/src/edu/stanford/nlp/trees/treebank/EnglishPTBTreebankCorrector.java b/src/edu/stanford/nlp/trees/treebank/EnglishPTBTreebankCorrector.java index e1ad9b5ea3..2d93578703 100644 --- a/src/edu/stanford/nlp/trees/treebank/EnglishPTBTreebankCorrector.java +++ b/src/edu/stanford/nlp/trees/treebank/EnglishPTBTreebankCorrector.java @@ -19,7 +19,7 @@ * You can use this class from {@link edu.stanford.nlp.trees.Treebanks} via a command like:

* {@code java edu.stanford.nlp.trees.Treebanks -correct -pennPrint LDC99T42-Treebank3/parsed/mrg/wsj 200-2199 > train-fixed} */ -public class EnglishPTBTreebankCorrector implements TreebankTransformer { +public class EnglishPTBTreebankCorrector implements TreeTransformer, TreebankTransformer { /** A logger for this class */ private static final Redwood.RedwoodChannels log = Redwood.channels(EnglishPTBTreebankCorrector.class); @@ -100,6 +100,11 @@ public EnglishPTBTreebankCorrector() { } } + @Override + public Tree transformTree(Tree t) { + return Tsurgeon.processPatternsOnTree(ops, t); + } + /** Fix all the English Penn Treebank errors, or at least some of them (!). */ @Override @@ -107,7 +112,7 @@ public MemoryTreebank transformTrees(Treebank tb) { MemoryTreebank mtb = new MemoryTreebank(tb.treeReaderFactory(), tb.encoding()); for (Tree t : tb) { - mtb.add(Tsurgeon.processPatternsOnTree(ops, t)); + mtb.add(transformTree(t)); } return mtb; }