Merge branch 'main' of github.com:Quantco/metalearners into narwhals

metalearners/drlearner.py

@@ -1,4 +1,4 @@
-# Copyright (c) QuantCo 2024-2024
+# Copyright (c) QuantCo 2024-2025
 # SPDX-License-Identifier: BSD-3-Clause
 
 

metalearners/rlearner.py

@@ -1,4 +1,4 @@
-# Copyright (c) QuantCo 2024-2024
+# Copyright (c) QuantCo 2024-2025
 # SPDX-License-Identifier: BSD-3-Clause
 
 

metalearners/tlearner.py

@@ -1,4 +1,4 @@
-# Copyright (c) QuantCo 2024-2024
+# Copyright (c) QuantCo 2024-2025
 # SPDX-License-Identifier: BSD-3-Clause
 
 from collections.abc import Mapping, Sequence

metalearners/xlearner.py

@@ -1,4 +1,4 @@
-# Copyright (c) QuantCo 2024-2024
+# Copyright (c) QuantCo 2024-2025
 # SPDX-License-Identifier: BSD-3-Clause
 
 

pixi.toml

@@ -5,47 +5,46 @@ platforms = ["linux-64", "osx-arm64", "osx-64", "win-64"]
 
 [tasks]
 postinstall = "pip install --no-build-isolation --no-deps --disable-pip-version-check -e ."
-coverage = "pytest tests --cov=metalearners --cov-report=xml --cov-report term-missing --color=yes"
-# Note that this task will fail on systems without bash.
-nb-convert = "nb-convert.sh"
-nb-mypy = "mypy docs/examples/*.py"
 
 [dependencies]
-pre-commit = "*"
-git_root = "*"
-pip = "*"
 python = ">=3.10"
-setuptools-scm = "*"
-setuptools = ">=61"
-ipykernel = "*"
-ipywidgets = "*"
-lime = "*"
-optuna = "*"
-nbconvert = "*"
-mypy = "*"
-pytest = ">=6"
-pytest-cov = "*"
-pytest-xdist = "*"
-polars = "*"
+shap = "*"
+matplotlib-base = "*"
+joblib = ">=1.2.0"
 # Remove the upper bound once this xgboost problem
 # has been fixed: https://github.com/dmlc/xgboost/issues/11093
 scikit-learn = ">=1.4,<1.6"
 numpy = "*"
 pandas = "*"
+narwhals = ">=1.25.0,<2"
+pyarrow = ">=19.0.0,<20"
+
+[host-dependencies]
+pip = "*"
+setuptools-scm = "*"
+setuptools = ">=61"
+
+[feature.dev.dependencies]
+git_root = "*"
+mypy = "*"
 typing-extensions = "*"
+ipykernel = "*"
+ipywidgets = "*"
 lightgbm = "*"
 xgboost = "*"
-glum = "*"
-shap = "*"
-matplotlib-base = "*"
-joblib = ">=1.2.0"
 lifelines = "*"
 onnx = "*"
 onnxmltools = "*"
 onnxruntime = "*"
 spox = "*"
-narwhals = ">=1.25.0,<2"
-pyarrow = ">=19.0.0,<20"
+optuna = "*"
+pytest = ">=6"
+pytest-cov = "*"
+pytest-xdist = "*"
+glum = "*"
+polars = "*"
+[feature.dev.tasks]
+coverage = "pytest tests --cov=metalearners --cov-report=xml --cov-report term-missing --color=yes"
 
 [feature.lint.dependencies]
 pre-commit = "*"
@@ -73,9 +72,13 @@ sphinx-autodoc-typehints = "*"
 myst-nb = "*"
 econml = ">=0.15"
 doubleml = "*"
+lime = "*"
 statsmodels = "*"
-
+nbconvert = "*"
 [feature.docs.tasks]
+# Note that this task will fail on systems without bash.
+nb-convert = "nb-convert.sh"
+nb-mypy = "mypy docs/examples/*.py"
 docs = "cd docs && make html "
 readthedocs = "rm -rf $READTHEDOCS_OUTPUT/html && cp -r docs/_build/html $READTHEDOCS_OUTPUT/html"
 
@@ -88,6 +91,7 @@ build = "python -m build"
 [feature.benchmark]
 platforms = ["linux-64", "osx-arm64", "osx-64"]
 [feature.benchmark.dependencies]
+pip = "*"
 git_root = "*"
 python = "3.10.*"
 scikit-learn = ">=1.3"
@@ -96,7 +100,6 @@ pandas = "*"
 lightgbm = "*"
 econml = ">=0.15"
 causalml = "*"
-pip = "*"
 tabulate = "*"
 
 [feature.benchmark.tasks]
@@ -111,13 +114,14 @@ python = "3.11.*"
 python = "3.12.*"
 
 [environments]
-py310 = ["py310"]
-py311 = ["py311"]
-py312 = ["py312"]
+default = ["dev"]
+py310 = ["py310", "dev"]
+py311 = ["py311", "dev"]
+py312 = ["py312", "dev"]
 
 lint = { features = ["lint"], no-default-feature = true }
 
-docs = ["docs"]
+docs = ["docs", "dev"]
 
 build = ["build"]