Method load

python_bindings/hnsw_bindings.cpp:128–141

            std::numeric_limits<float>::infinity()
        );
        auto ids_buf = ids.mutable_unchecked<2>();
        auto dists_buf = dists.mutable_unchecked<2>();

        std::vector<std::vector<std::pair<float, rabitqlib::PID>>> results = index_->search(
            query_array.data(),
            static_cast<size_t>(query_array.shape(0)),
            k,
            ef,
            num_threads
        );

        for (ssize_t i = 0; i < static_cast<ssize_t>(results.size()); ++i) {
            for (ssize_t j = 0; j < static_cast<ssize_t>(std::min<size_t>(
                                        k, results[static_cast<size_t>(i)].size()
                                    ));
                 ++j) {
                ids_buf(i, j) =