add pytorch_frame and pytorch_geometric to dependencies

kumo-ai · Jul 11, 2024 · 46bd89d · 46bd89d
1 parent 90d562a
commit 46bd89d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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]