This does not work and I spent the afternoon trying to understand why...

TypeError: 'consume' operand is not isolated
Exception ignored in: 'KDTree.query'
Traceback (most recent call last):
  File "/home/jjerphan/dev/kdtree/debug.py", line 18, in <module>
      tree.query(Y, knn_indices, knn_distances)
      TypeError: 'consume' operand is not isolated