pyproject.toml

[tool.poetry]
name = "age-related-conditions"
version = "0.1.0"
description = "Kaggle competition: Identifying age-related conditions"
authors = ["beekill <nguyenmbquan95@gmail.com>"]
license = "MIT"
readme = "README.md"
packages = [{ include = "age_related_conditions" }]

[tool.poetry.dependencies]
python = ">=3.10,<3.12"
numpy = "^1.24.3"
matplotlib = "^3.7.1"
scikit-learn = "^1.2.2"
pandas = "^2.0.1"
seaborn = "^0.12.2"
scipy = "^1.10.1"
xgboost = "^1.7.5"
lightgbm = "^3.3.5"
catboost = "^1.2"
arviz = "^0.15.1"
funsor = "^0.4.5"
pyro-ppl = "^1.8.5"
tabpfn = "^0.1.9"
imbalanced-learn = "^0.11.0"
itables = "^1.5.3"
tensorboard = "^2.13.0"

[tool.poetry.group.cpu]
optional = true

[tool.poetry.group.cpu.dependencies]
torch = { version = "^2.0.1+cpu", source = "torch-cpu" }
torchvision = { version = "^0.15.2+cpu", source = "torch-cpu" }
torchaudio = { version = "^2.0.2+cpu", source = "torch-cpu" }
numpyro = "^0.12.1"

[tool.poetry.group.dev.dependencies]
jupyterlab = "^4.0.0"
jupytext = "^1.14.5"
kaggle = "^1.5.13"
ipywidgets = "^8.0.6"
itables = "^1.5.2"


[tool.poetry.group.gpu]
optional = true

[tool.poetry.group.gpu.dependencies]
torch = {version = "^2.0.1+cu118", source = "torch-gpu"}
torchvision = {version = "^0.15.2+cu118", source = "torch-gpu"}
torchaudio = {version = "^2.0.2+cu118", source = "torch-gpu"}
numpyro = {extras = ["cuda"], version = "^0.12.1"}
jax = {extras = ["cuda11-pip"], version = "^0.4.13"}

[[tool.poetry.source]]
name = "torch-cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "supplemental"


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


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


[[tool.poetry.source]]
name = "jax-gpu"
url = "https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"
priority = "supplemental"

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