diff --git a/pyproject.toml b/pyproject.toml
index cffa7b8..dc22617 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -52,6 +52,8 @@ full=[
     "mpmath==1.3.0",
     "datasets",
     "torchmetrics",
+    "pytorch_frame>=0.2.3",
+    "torch_geometric",
 ]
 
 [project.urls]