pyproject.toml

[project]
name = "tabm"
version = "0.0.1.dev0"
# PyPI packages.
dependencies = [
    "delu==0.0.25",
    "rtdl_num_embeddings==0.0.11",
    "rtdl_revisiting_models==0.0.2",
]

[tool.pixi.project]
platforms = ["linux-64", "osx-arm64", "osx-64"]
channels = ["nvidia", "conda-forge", "pytorch", "nodefaults"]

# Conda packages.
[tool.pixi.dependencies]
# Main (experiments can directly import these dependencies)
loguru = "==0.7.2"
numpy = "==1.26.3"
optuna = "==3.3.0"
pandas = "==2.1.1"
polars = "==1.5.0"
python = "==3.10.12"
pytorch = { version = "==2.0.1", channel = "pytorch" }
scikit-learn = "==1.3.1"
scipy = "==1.11.3"
tensorboard = "==2.15.2"
tomli = "==2.0.1"
tomli-w = "==1.0.0"
tqdm = "==4.66.1"
typing-extensions = "==4.8.0"
# Development (results do not depend on these packages)
bokeh = "==3.3.4"
holoviews = "==1.18.3"
hvplot = "==0.10.0"
ipython = "==8.16.1"
ipywidgets = "==8.1.3"
jupyterlab = "==4.1.1"
mypy = "==1.7.1"
panel = "==1.3.6"
pip = "==23.3.1"
psutil = "==5.9.5"
python-dotenv = "==1.0.1"
ruff = "==0.6.0"
seaborn = "==0.13.2"

[tool.pixi.pypi-options]
index-url = "https://pypi.org/simple"

[tool.pixi.feature.cuda]
system-requirements = { libc = { family = "glibc", version = "2.27" } }
# In fact, it should be this:
# system-requirements = { cuda = "11.8", libc = { family = "glibc", version = "2.27" } }
# But the "cuda" requirement breaks the project on macOS.

# (As of pixi==0.29.0) The ".target.linux-64" part is a workaround for projects
# with multiple platforms AND platform-specific dependencies (e.g. cuda) AND PyPI dependencies.
# Details: https://github.com/prefix-dev/pixi/issues/1051#issuecomment-2088820974
[tool.pixi.feature.cuda.target.linux-64.dependencies]
pytorch-cuda = { version = "==11.8", channel = "pytorch" }

[tool.pixi.environments]
default = { solve-group = "default" }
cuda = { features = ["cuda"], solve-group = "default" }

[tool.mypy]
check_untyped_defs = true
ignore_missing_imports = true

[tool.ruff]
target-version = "py310"
line-length = 88

[tool.ruff.lint]
extend-select = ["RUF", "UP", "E501", "E731"]

[tool.ruff.format]
quote-style = "single"