From c0582def91b8b39fbae2901bf5864861ea273fa0 Mon Sep 17 00:00:00 2001
From: luyuncheng <luyuncheng@bytedance.com>
Date: Mon, 27 Nov 2023 16:20:15 +0800
Subject: [PATCH] Fix is_trained in IndexNSGSQ

---
 faiss/IndexNSG.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/faiss/IndexNSG.cpp b/faiss/IndexNSG.cpp
index 23710c4f59..9652266c43 100644
--- a/faiss/IndexNSG.cpp
+++ b/faiss/IndexNSG.cpp
@@ -309,7 +309,7 @@ IndexNSGSQ::IndexNSGSQ(
         int M,
         MetricType metric)
         : IndexNSG(new IndexScalarQuantizer(d, qtype, metric), M) {
-    is_trained = false;
+    is_trained = this->storage->is_trained;
     own_fields = true;
 }