summaryrefslogtreecommitdiff
path: root/python/src/kbest.pxd
diff options
context:
space:
mode:
Diffstat (limited to 'python/src/kbest.pxd')
-rw-r--r--python/src/kbest.pxd8
1 files changed, 6 insertions, 2 deletions
diff --git a/python/src/kbest.pxd b/python/src/kbest.pxd
index e339714a..cef42bd3 100644
--- a/python/src/kbest.pxd
+++ b/python/src/kbest.pxd
@@ -1,16 +1,20 @@
from libcpp.vector cimport vector
from utils cimport WordID
-cimport hypergraph
+from hypergraph cimport Hypergraph
cdef extern from "decoder/viterbi.h":
cdef cppclass ESentenceTraversal:
pass
cdef cppclass ETreeTraversal:
pass
+ cdef cppclass FTreeTraversal:
+ pass
+ cdef cppclass FeatureVectorTraversal:
+ pass
cdef extern from "decoder/kbest.h" namespace "KBest":
cdef cppclass KBestDerivations[T, Traversal]:
cppclass Derivation:
T _yield "yield"
- KBestDerivations(hypergraph.Hypergraph hg, unsigned k)
+ KBestDerivations(Hypergraph& hg, unsigned k)
Derivation* LazyKthBest(unsigned v, unsigned k)