fix bug

mldrugdiscovery · Oct 24, 2019 · 04dcaa0 · 04dcaa0
1 parent 16ca7da
commit 04dcaa0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python_bindings/nmslib.cc b/python_bindings/nmslib.cc
@@ -199,16 +199,17 @@ 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 = res.begin();
+	int i=0;
 	while (it != res.end())
 	{
-		ids.mutable_at(i) = it->id();
+		ids.mutable_at(i) = (*it)->id();
 		//distances.mutable_at(size) = res->at(i)->id();
-		it++;
+		it++; i++;
 	}
     return py::make_tuple(ids, distances);
   }