fix hgraph deserialize bug (#417)

Signed-off-by: LHT129 <[email protected]>
antgroup · Feb 18, 2025 · 7b1ebaa · 7b1ebaa
1 parent 6e0330d
commit 7b1ebaa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/algorithm/hgraph.cpp b/src/algorithm/hgraph.cpp
@@ -586,7 +586,8 @@ HGraph::Deserialize(StreamReader& reader) {
     for (uint64_t i = 0; i < this->max_level_; ++i) {
         this->route_graphs_[i]->Deserialize(reader);
     }
-    resize(max_capacity_);
+    this->neighbors_mutex_->Resize(max_capacity_);
+    pool_ = std::make_shared<hnswlib::VisitedListPool>(max_capacity_, allocator_);
 }
 
 void

diff --git a/tests/test_hgraph.cpp b/tests/test_hgraph.cpp
@@ -37,7 +37,7 @@ class HgraphTestIndex : public fixtures::TestIndex {
 
     static fixtures::TempDir dir;
 
-    constexpr static uint64_t base_count = 1000;
+    constexpr static uint64_t base_count = 1200;
 
     constexpr static const char* search_param_tmp = R"(
         {{