From b9561473b598b0b2c750fc911f2cb3f2f51d747b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20Aum=C3=BCller?= <maau@itu.dk>
Date: Thu, 3 May 2018 13:30:37 +0200
Subject: [PATCH] Correctly set number of probes in FAISS.

---
 ann_benchmarks/algorithms/faiss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ann_benchmarks/algorithms/faiss.py b/ann_benchmarks/algorithms/faiss.py
index 6bf975c44..14b414b70 100644
--- a/ann_benchmarks/algorithms/faiss.py
+++ b/ann_benchmarks/algorithms/faiss.py
@@ -57,7 +57,7 @@ def fit(self, X):
 
     def set_query_arguments(self, n_probe):
         self._n_probe = n_probe
-        self._index.n_probe = self._n_probe
+        self._index.nprobe = self._n_probe
 
     def query(self, v, n):
         if self._metric == 'angular':