索引越界问题！

[xxx2974]

作者你好，我在测试期间对knn graph进行构建时，利用类似于下面的语句构建：
result, dists = flann.nn(dataset, testset, 201, algorithm="kmeans", branching=32, iterations=7, checks=16)，但是我使用的dataset, testset,两个数据的大小是不一样的，dataset包含了testset。
在测试时出现了以下问题：
InsexError: index 6209 is out of bounds for axis 0 with size 3368.
具体报错代码是：hops[-1].update(set(self.knn_graph[h][1:self.k_at_hop[d]+1]))

请问这个问题可以解决吗？谢谢！