Update requirements.txt

requirements.txt

@@ -1,13 +1,8 @@
-# --- PyTorch (CPU) ---
---index-url https://download.pytorch.org/whl/cpu
-torch==2.2.2
-torchvision==0.17.2
-torchaudio==2.2.2
+# --- Torch (CPU) ---
+torch==2.2.0
 
-# --- PyG core ---
-torch-geometric==2.5.3
-
-# --- PyG extension wheels (match TORCH=2.2.0, CUDA=cpu) ---
+# --- PyG + compiled ops (CPU wheels) ---
+torch-geometric>=2.6.0
 -f https://data.pyg.org/whl/torch-2.2.0+cpu.html
 pyg-lib
 torch-scatter
@@ -15,7 +10,7 @@ torch-sparse
 torch-cluster
 torch-spline-conv
 
-# --- the rest of your deps ---
+# --- Your existing deps (keep as you had) ---
 numpy>=1.24
 pandas>=2.0
 scikit-learn>=1.3