update

Author: rusty1s

.github/workflows/building.yml

@@ -10,7 +10,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-22.04, macos-14, windows-2022, ubuntu-22.04-arm]
+        os: [windows-2022]
+        # os: [ubuntu-22.04, macos-14, windows-2022, ubuntu-22.04-arm]
         python-version: ['3.10', '3.11', '3.12', '3.13', '3.14']
         torch-version: [2.9.0, 2.10.0]
         cuda-version: ['cpu', 'cu126', 'cu128', 'cu130']

torch_cluster/nearest.py

@@ -120,5 +120,7 @@ def nearest(
             y = torch.cat([y, 2 * D * batch_y.view(-1, 1).to(y.dtype)], -1)
 
         return torch.from_numpy(
-            scipy.cluster.vq.vq(x.detach().cpu(),
-                                y.detach().cpu())[0]).to(torch.long)
+            scipy.cluster.vq.vq(
+                x.detach().cpu().float(),
+                y.detach().cpu().float(),
+            )[0]).to(torch.long)