diff --git a/pyproject.toml b/pyproject.toml
index cf48ba9..b05e178 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,9 +49,9 @@ full=[
     "datasets",
     "torchmetrics",
     "pytorch_frame>=0.2.3",
+    "pyg_lib",
     "torch_geometric",
     "relbench==1.0.0rc1",
-    "torch-sparse",
 ]
 
 [project.urls]