Merge pull request #1 from thunderock/setup_geometric

Setup geometric

Author: thunderock

Makefile

@@ -28,5 +28,6 @@ setup_all: clean
 .PHONY: format
 format:
 	@echo "formatting..."
+	@poetry install --only lint
 	@poetry run black .
 	@poetry run pre-commit run --all-files

poetry.lock

@@ -32,6 +32,12 @@ pytorch-lightning = "^2.1.3"
 traittypes = "^0.2.1"
 traitlets = "^5.9.0"
 datasets = "^2.17.1"
+torch-cluster = {url = "https://data.pyg.org/whl/torch-2.1.0%2Bcu121/torch_cluster-1.6.2%2Bpt21cu121-cp310-cp310-linux_x86_64.whl"}
+pyg-lib = {url = "https://data.pyg.org/whl/torch-2.1.0%2Bcu121/pyg_lib-0.4.0%2Bpt21cu121-cp310-cp310-linux_x86_64.whl"}
+torch-scatter = {url = "https://data.pyg.org/whl/torch-2.1.0%2Bcu121/torch_scatter-2.1.2%2Bpt21cu121-cp310-cp310-linux_x86_64.whl"}
+torch-sparse = {url = "https://data.pyg.org/whl/torch-2.1.0%2Bcu121/torch_sparse-0.6.18%2Bpt21cu121-cp310-cp310-linux_x86_64.whl"}
+torch-spline-conv = {url = "https://data.pyg.org/whl/torch-2.1.0%2Bcu121/torch_spline_conv-1.2.2%2Bpt21cu121-cp310-cp310-linux_x86_64.whl"}
+torch-geometric = "^2.5.0"
 
 # [[tool.poetry.source]]
 # name = "torch-wheels"