1 files changed, 16 insertions, 0 deletions
diff --git a/python/src/lattice.pxd b/python/src/lattice.pxd
new file mode 100644
index 00000000..bdfaba80
--- /dev/null
+++ b/python/src/lattice.pxd
@@ -0,0 +1,16 @@
+from libcpp.vector cimport vector
+from utils cimport WordID
+
+cdef extern from "decoder/lattice.h":
+    cdef cppclass LatticeArc:
+        WordID label
+        double cost
+        int dist2next
+        LatticeArc()
+        LatticeArc(WordID w, double c, int i)
+
+    cdef cppclass Lattice: # (vector[vector[LatticeArc]])
+        Lattice()
+        Lattice(unsigned t)
+        Lattice(unsigned t, vector[LatticeArc] v)
+        bint IsSentence()