1 files changed, 4 insertions, 27 deletions
diff --git a/klm/search/edge.hh b/klm/search/edge.hh
index 4d2a5cbf..77ab0ade 100644
--- a/klm/search/edge.hh
+++ b/klm/search/edge.hh
@@ -11,33 +11,6 @@
 
 namespace search {
 
-class Edge {
-  public:
-    Edge() {
-      end_to_ = to_;
-    }
-
-    Rule &InitRule() { return rule_; }
-
-    void Add(Vertex &vertex) {
-      assert(end_to_ - to_ < kMaxArity);
-      *(end_to_++) = &vertex;
-    }
-
-    const Vertex &GetVertex(std::size_t index) const {
-      return *to_[index];
-    }
-
-    const Rule &GetRule() const { return rule_; }
-
-  private:
-    // Rule and pointers to rule arguments.  
-    Rule rule_;
-
-    Vertex *to_[kMaxArity];
-    Vertex **end_to_;
-};
-
 struct PartialEdge {
   Score score;
   // Terminals
@@ -45,6 +18,10 @@ struct PartialEdge {
   // Non-terminals
   PartialVertex nt[kMaxArity];
 
+  const lm::ngram::ChartState &CompletedState() const {
+    return between[0];
+  }
+
   bool operator<(const PartialEdge &other) const {
     return score < other.score;
   }