mitra-finetune / pyproject.toml
yft's picture
Cheaper fine-tuning loop and prediction (v0.3.0)
b4701e8 verified
Raw History Blame Contribute Delete
601 Bytes
[build-system]
requires = ["setuptools>=77"]
build-backend = "setuptools.build_meta"
[project]
name = "mitra-finetune"
version = "0.3.0"
description = "Fine-tuning for pretrained Mitra tabular foundation-model checkpoints"
readme = "README.md"
requires-python = ">=3.11"
license = "Apache-2.0"
license-files = ["LICENSE"]
authors = [{ name = "Boran Han" }]
dependencies = [
"numpy>=1.24",
"scikit-learn>=1.4",
"torch>=2.1",
]
[project.optional-dependencies]
full = ["autogluon.tabular[mitra]>=1.6", "tabarena>=0.1.0"]
dev = ["pytest>=7"]
[tool.setuptools.packages.find]
where = ["src"]