Support arbitrary arity

Also, some windows fixes in util.

1 files changed, 2 insertions, 2 deletions

diff --git a/search/edge_generator.hh b/search/edge_generator.hh
index 43f257929..582c78b7b 100644
--- a/search/edge_generator.hh
+++ b/search/edge_generator.hh
@@ -22,7 +22,7 @@ class EdgeGenerator {
     EdgeGenerator() {}
 
     PartialEdge AllocateEdge(Arity arity) {
-      return partial_edge_pool_.Allocate(arity);
+      return PartialEdge(partial_edge_pool_, arity);
     }
 
     void AddEdge(PartialEdge edge) {
@@ -47,7 +47,7 @@ class EdgeGenerator {
     }
 
   private:
-    PartialEdgePool partial_edge_pool_;
+    util::Pool partial_edge_pool_;
 
     typedef std::priority_queue<PartialEdge> Generate;
     Generate generate_;