diff options
Diffstat (limited to 'python/src/kbest.pxd')
| -rw-r--r-- | python/src/kbest.pxd | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/python/src/kbest.pxd b/python/src/kbest.pxd index e339714a..f34ac5ed 100644 --- a/python/src/kbest.pxd +++ b/python/src/kbest.pxd @@ -1,16 +1,18 @@ 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 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) |