fix bug

mldrugdiscovery · Oct 24, 2019 · 16ca7da · 16ca7da
1 parent 129b72e
commit 16ca7da
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python_bindings/nmslib.cc b/python_bindings/nmslib.cc
@@ -199,12 +199,12 @@ struct IndexWrapper {
 
   py::object convertResult_rq(std::set<const Object*> res) {
     // Create numpy arrays for the output
-    size_t size = res->Size();
+    size_t size = res.Size();
     py::array_t<int> ids(size);
     py::array_t<dist_t> distances(size);
 
-	std::set<const Object*>::iterator it = setOfStr.begin();
-	while (it != setOfStr.end())
+	std::set<const Object*>::iterator it = res.begin();
+	while (it != res.end())
 	{
 		ids.mutable_at(i) = it->id();
 		//distances.mutable_at(size) = res->at(i)->id();