pixi.toml

[workspace]
channels = ["conda-forge"]
name = "glum"
platforms = ["linux-64", "osx-arm64", "win-64"]

[tasks]
postinstall = "pip install --no-build-isolation --no-deps --disable-pip-version-check -e ."

[feature.test.tasks]
store-benchmark-golden-master = { cmd = "python tests/glm/test_benchmark_golden_master.py", env = { PYTHONPATH = "." } }
store-golden-master = { cmd = "python tests/glm/test_golden_master.py", env = { PYTHONPATH = "." } }
test = { cmd = "pytest tests/glm --doctest-modules src/glum", env = { PYTHONPATH = "." } }

[feature.docs.tasks]
make-docs = "cd docs && make html"
readthedocs = { cmd = "rm -rf $READTHEDOCS_OUTPUT/html && cp -r docs/_build/html $READTHEDOCS_OUTPUT/html", depends-on = [
    "make-docs",
] }
serve-docs = { cmd = "python -m http.server --directory docs/_build/html", depends-on = [
    "make-docs",
] }

[feature.benchmark.tasks]
run-benchmarks = { cmd = "python glum_benchmarks/run_benchmarks.py", env = { PYTHONPATH = "." } }
test-benchmarks = { cmd = "pytest glum_benchmarks/tests", env = { PYTHONPATH = "." } }

[feature.lint.tasks]
pre-commit-install = "pre-commit install"
pre-commit-run = "pre-commit run -a"

[feature.no-tabmat.tasks.postinstall-glum-tabmat]
cmd = """
pip install --no-build-isolation --no-deps --disable-pip-version-check -e ../tabmat
&& pip install --no-build-isolation --no-deps --disable-pip-version-check -e .
"""

[feature.build-tabmat.tasks.install-nightlies]
cmd = """
    PRE_WHEELS="https://pypi.anaconda.org/scientific-python-nightly-wheels/simple/"
    && pip install --pre --no-deps --only-binary :all: --upgrade --timeout=60 -i $PRE_WHEELS pandas
    && pip install --pre --no-deps --only-binary :all: --upgrade --timeout=60 -i $PRE_WHEELS scikit-learn
    && pip install --pre --no-deps --only-binary :all: --upgrade --timeout=60 -i $PRE_WHEELS statsmodels
    && pip install --pre --no-deps --only-binary :all: --upgrade --timeout=60 -i https://pypi.fury.io/arrow-nightlies/ pyarrow
    && pip install --no-build-isolation --no-deps git+https://github.com/Quantco/tabmat
"""

[dependencies]
c-compiler = "*"
cxx-compiler = "*"
cython = "*"
formulaic = ">=1.2.0"
narwhals = ">=2.0.0"
numexpr = "*"
numpy = ">=1.24"
packaging = "*"
pandas = ">=1.4"
pip = "*"
pyarrow = "*"
python = ">=3.10"
scikit-learn = ">=1.1.0"
scipy = ">=1.8.0"
setuptools = ">=61"
setuptools-scm = ">=8.1"
tabmat = ">=4.2.0"
tqdm = "*"
wheel = "*"

[feature.no-tabmat.dependencies]
formulaic = "*"
numexpr = "*"
pandas = ">=1.1"
pyarrow = "*"
scikit-learn = ">=0.23"
scipy = "*"
tqdm = "*"

[feature.test.dependencies]
attrs = "*"
click = "*"
git_root = "*"
mypy = "*"
polars = "*"
psutil = "*"
pytest = "*"
pytest-xdist = "*"
statsmodels = "*"

[feature.dev.dependencies]
ipdb = "*"
ipykernel = "*"
ipython = "*"
line_profiler = "*"
memory_profiler = "*"

[feature.lint.dependencies]
cython-lint = "*"
pre-commit = "*"
pre-commit-hooks = "*"
ruff = "*"

[feature.docs.dependencies]
jinja2 = "*"
jupyterlab = "*"
jupytext = "*"
make = "*"
matplotlib-base = "*"
nbclassic = ">=0.2.8"
nbsphinx = ">=0.8.3"
sphinx = ">=3.5.3,!=8.2"
sphinx_rtd_theme = "*"
sphinxcontrib-apidoc = "*"
sphinxext-altair = "*"

[feature.tutorial.dependencies]
altair = "*"            # used in docs/tutorials/rossman
dask-ml = ">=2022.5.27" # used in tutorials rossman and insurance
jupyterlab = "*"
libpysal = "*"          # used in docs/tutorials/regularization_housing_data
lifelines = "*"         # used in docs/tutorials/cox_model
openml = "*"            # used for downloading datasets in the tutorials.
shapely = "*"           # used in docs/tutorials/regularization_housing_data

[feature.benchmark.dependencies]
attrs = "*"
git_root = "*"
matplotlib-base = "*"
openjdk = "*"
pydantic = "*"
pytest = "*"
r-base = "*"
r-glmnet = "*"
rich = "*"
rpy2 = "*"
"ruamel.yaml" = "*"
skglm = "*"

[feature.benchmark.pypi-dependencies]
celer = "*"
h2o = "*"   # Latest version from PyPI

[feature.benchmark.target.win-64.dependencies]
blas = { build = "*mkl" }
[feature.benchmark.target.linux-64.dependencies]
blas = { build = "*mkl" }

[feature.build-tabmat.dependencies]
c-compiler = "*"
cxx-compiler = "*"
cython = "!=3.0.4"
make = "*"
mako = "*"
narwhals = ">=2.0.0"
pip = "*"
setuptools-scm = "*"
xsimd = "<11|>12.1"
[feature.build-tabmat.target.unix.dependencies]
jemalloc-local = "*"

[feature.py310.dependencies]
python = "3.10.*"
[feature.py311.dependencies]
python = "3.11.*"
[feature.py312.dependencies]
python = "3.12.*"
[feature.py313.dependencies]
python = "3.13.*"

[feature.oldies.dependencies]
formulaic = "1.2.*"
narwhals = "2.0.*"
numpy = "1.24.*"
pandas = "1.4.*"
python = "3.10.*"
scikit-learn = "==1.1.0"
scipy = "1.8.*"
tabmat = "==4.2.0"

[environments]
benchmark = ["benchmark"]
default = ["dev", "test"]
docs = ["docs", "tutorial"]
glum-tabmat = { features = [
    "build-tabmat",
    "dev",
    "no-tabmat",
    "test",
], no-default-feature = true }
lint = { features = ["lint"], no-default-feature = true }
nightly = ["build-tabmat", "py313", "test"]
oldies = ["oldies", "test"]
py310 = ["py310", "test"]
py311 = ["py311", "test"]
py312 = ["py312", "test"]
py313 = ["py313", "test"]