fix pyproject.toml

GFNOrg · Jan 13, 2025 · 9bbc48d · 9bbc48d
1 parent 12c49b7
commit 9bbc48d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,6 +45,7 @@ wandb = { version = "*", optional = true }
 scikit-learn = {version = "*", optional = true }
 scipy = { version = "*", optional = true }
 matplotlib = { version = "*", optional = true }
+torch_geometric = { version = "*", optional = true }
 
 [tool.poetry.extras]
 dev = [