[tool.poetry]
name = "reproducible"
version = "0.0.1"
description = "Reproducible Machine Learning Environment"
authors = ["Leni Aniva <v@leni.sh>"]
readme = "README.md"
include = []
packages = [{ include = "reproducible" }]

[tool.poetry.scripts]
reproducible = "reproducible.util:main"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.poetry.dependencies]
python = "^3.11"
numpy = "^1.26"
#torch = { url = "https://download.pytorch.org/whl/cu118/torch-2.2.1+cu118-cp311-cp311-linux_x86_64.whl" }
torch = { version = "^2.2.0", source = "pytorch-cu118" }
torch-geometric = "^2.5.2"
#pyg-lib = { git = "https://github.com/pyg-team/pyg-lib.git", tag = "0.4.0" }
#torch-cluster = { version = "1.6.3" }
#torch-scatter = { version = "2.1.2" }
#torch-sparse = { version = "0.6.18" }
#torch-spline-conv = { version = "1.2.2" }

#pyg-lib = { version = "0.4.0", source = "pyg-torch220-cu118" }
#torch-cluster = { version = "1.6.3", source = "pyg-torch220-cu118" }
#torch-scatter = { version = "2.1.2", source = "pyg-torch220-cu118" }
#torch-sparse = { version = "0.6.18", source = "pyg-torch220-cu118" }
#torch-spline-conv = { version = "1.2.2", source = "pyg-torch220-cu118" }

[[tool.poetry.source]]
name = "pytorch-cu118"
url = "https://download.pytorch.org/whl/cu118"
priority = "explicit"

#[[tool.poetry.source]]
#name = "pyg-torch220-cu118"
#url = "git+https://github.com/pyg-team/pyg-lib.git"
#url = "https://data.pyg.org/whl/torch-2.2.0+cu118.html"
#priority = "explicit"

#[[tool.poetry.source]]
#name = "PyPI"
#priority = "primary"