pyproject.toml

[tool.poetry]
name = "py-tgb"
version = "2.3.0"
description = "Temporal Graph Benchmark project repo"
authors = ["shenyang Huang <shenyang.huang@mail.mcgill.ca>", "Julia Gastinger", "Farimah Poursafaei", "Emanuele Rossi <emanuele.rossi1909@gmail.com>", "Jacob Danovitch <jacob.danovitch@mila.quebec>"]
readme = "README.md"
packages = [{include = "tgb"}]

[tool.poetry.dependencies]
python = "^3.9"
torch-geometric = {version = "^2.3.0", optional = true}
tqdm = "^4.65.0"
numpy = "^2.0.2"
clint = "^0.5.1"
requests = "^2.28.2"
pandas = ">=2.2.3"
scikit-learn = "^1.2.2"

[tool.poetry.extras]
torch = ["torch-geometric"]

[tool.poetry.group.dev.dependencies]
mkdocs = "^1.4.3"
mkdocs-material = "^9.1.15"
mkdocstrings-python = "^1.1.2"
mkdocs-jupyter = "^0.24.1"
poetry = "^1.5.1"

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

[tool.pytest.ini_options]
testpaths = ["test"]
markers = [
    "network: test performs live network access (deselect with '-m \"not network\"'); opt in with --run-network or TGB_RUN_NETWORK=1",
]