fix: updated poetry

akhilpandey95 · Oct 19, 2023 · bcd072c · bcd072c
1 parent 5a6d3fc
commit bcd072c
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 29 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,13 +8,19 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
-torch = "^2.1.0"
 numpy = "^1.26.1"
-torch-geometric = "^2.4.0"
 scikit-learn = "^1.3.1"
 pdoc3 = "^0.10.0"
 pytest = "^7.4.2"
+torch-geometric = "^2.4.0"
+torch = "^2.1.0+cu118"
+
 
+[[tool.poetry.source]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cu118"
+default = false
+secondary = false
 
 [build-system]
 requires = ["poetry-core"]