JeppeLS

I'm still experiencing this bug on these versions ``` torch 2.2.2 torch_cluster 1.6.3+pt22cu121 torch_geometric 2.5.2 torch_sparse 0.6.18+pt22cu121 ``` A small example to reproduce ```python import torch from torch_geometric.nn import knn...