Thank you for taking the time to file a bug report.

Thank you for taking the time to file a bug report.: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). I am having an issue for a specific usage of the count neighbors method of a KDTree. With the new "weights" argument in 1.6.0, this error occurs when counting…

The task

I am having an issue for a specific usage of the count_neighbors method of a KDTree. With the new "weights" argument in 1.6.0, this error occurs when counting neighbors between two different arrays A and B, with their respective weight arrays in a tuple for the "weights" arg :

Part of FineEnvs/MiMo-V2.6-RL-harbor-code.