machine-learning/graph-neural-network/pyproject.toml
[tool.poetry]
package-mode = false
[tool.poetry.dependencies]
python = "3.8.x"
numpy = "1.21.6"
ogb = "1.3.6"
torch = "1.13.1"
torch-geometric = "2.3.1"
tqdm = "4.66.5"
wandb = "0.17.9"
[tool.poetry.group.dev.dependencies]
poethepoet = "0.19.0"
pytest = "7.4.4"
pytest-cov = "4.1.0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.poe.tasks]
dev = "python src/main.py"
test = "pytest --verbose --verbose"
test-coverage = "pytest --cov=. --cov-report=xml"