tabicl/pyproject.toml at main · ManuelLecha/tabicl · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
[project]
name = "tabicl"
dynamic = ["version"]
description = "TabICL: A state-of-the-art tabular foundation model"
readme = "README.md"
authors = [
    { name = "Jingang Qu" },
    { name = "David Holzmüller" } ,
    { name = "Marine Le Morvan" } ,
    { name = "Gaël Varoquaux" } ,
]
license = { file = "LICENSE" }
requires-python = ">=3.9"
keywords = ["tabular data", "foundation model", "TabICL", "in-context learning"]
classifiers = [
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "Topic :: Scientific/Engineering",
    "Development Status :: 4 - Beta",
    "Programming Language :: Python",
    'Programming Language :: Python :: 3',
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
]
dependencies = [
    "torch>=2.2",  # to enjoy Flash Attention v2
    "scikit-learn>=1.3.0",
    "numpy",
    "scipy",
    "einops>=0.7",  # this is only used for rope, so it could be replaced
    "psutil",  # used for memory calculation at inference
    "tqdm>=4.64.0",
    "huggingface-hub",
]

[project.optional-dependencies]
pretrain = [
    "transformers",  # only used in optim for schedules  (maybe implement them natively to remove the dependency?)
    "xgboost",  # used in the TabICLv1 prior
    "wandb",  # logging in pre-training
    "joblib",  # parallelize dataset generation
    "pandas",  # used in tests / eval
]

[build-system]
requires = ["hatchling>=1.26.1"]  # https://github.com/pypa/hatch/issues/1818
build-backend = "hatchling.build"

[tool.hatch.version]
path = "src/tabicl/__about__.py"

[tool.hatch.build.targets.sdist]
only-include = ["src/tabicl", "figures", "LICENSE", "README.md"]

[tool.hatch.build.targets.wheel]
packages = ["src/tabicl"]

[tool.hatch.envs.default]
installer = "uv"
features = []

[tool.hatch.envs.hatch-test]
default-args = ["tests"]
installer = "uv"
features = []

[tool.pytest.ini_options]
pythonpath = "src"  # to allow importing tabicl instead of src.tabicl

[tool.hatch.envs.types]
extra-dependencies = [
    "mypy>=1.0.0",
]
[tool.hatch.envs.types.scripts]
check = "mypy --install-types --non-interactive {args:tabicl tests}"

[tool.coverage.run]
source = ["src"]
branch = true
parallel = true
omit = [
    "src/tabicl/__about__.py",
]

[tool.coverage.paths]
tabicl = ["src"]

[tool.coverage.report]
exclude_lines = [
    "no cov",
    "if __name__ == .__main__.:",
    "if TYPE_CHECKING:",
]