causalis 1.0.3

A Python toolkit for causal inference and experimentation

Causalis

Robust causal inference for experiments and observational studies in Python, organized around scenarios (e.g., Classic RCT, CUPED, Unconfoundedness) with a consistent fit() → estimate() workflow.

• 📚 Documentation & notebooks: https://causalis.causalcraft.com/
• 🔎 API reference: https://causalis.causalcraft.com/api-reference

Why Causalis?

Causalis focuses on:

• Scenario-first workflows (you pick the study design; Causalis provides best-practice defaults).
• Extensive robustness tests that reveal issues in the study design or model specification
• Pydantic data contracts
• An advanced DGP (Data Generating Process) with heterogeneous treatment effects, latent variables, and correlated confounders
• A website with notebooks based on real-world cases

Installation

Recommended

pip install causalis

Quickstart: Classic RCT (difference in means + inference)

from causalis.dgp import generate_classic_rct_26
from causalis.scenarios.classic_rct import DiffInMeans, check_srm

# Synthetic RCT data as a validated CausalData object
data = generate_classic_rct_26(seed=42, return_causal_data=True)

# Optional: Sample Ratio Mismatch check
srm = check_srm(data, target_allocation={0: 0.5, 1: 0.5}, alpha=1e-3)
print("SRM detected?", srm.is_srm, "p=", srm.p_value, "chi2=", srm.chi2)

# Estimate treatment effect with t-test inference (or bootstrap / conversion_ztest)
result = DiffInMeans().fit(data).estimate(method="ttest", alpha=0.05)
result.summary()

Quickstart: Observational study (Unconfoundedness / DML IRM)

from causalis.scenarios.unconfoundedness.dgp import generate_obs_hte_26
from causalis.scenarios.unconfoundedness import IRM
from causalis.data_contracts import CausalData

causaldata = generate_obs_hte_26(return_causal_data=True, include_oracle=False)

from causalis.scenarios.unconfoundedness import IRM

model = IRM().fit(causaldata)
result = model.estimate(score='ATTE')
result.summary()

Pick your scenario

Scenario	Estimator	Assumptions
Classic RCT	Difference in means (ttest, ztest, welch_permutation_t_test)	Random assignment, no sample ratio mismatch, SUTVA
CUPED	CUPED-adjusted difference in means with Lin specification	Random assignment, no sample ratio mismatch, SUTVA, valid pre-period metrics
Unconfoundedness	DML IRM	Unconfoundedness, Overlap, SUTVA, No leakage, Score stability
GATE	DML IRM (GATE and GATET)	Same assumptions as unconfoundedness, plus meaningful pre-specified or validated subgroup definitions.
Multi Unconfoundedness	Multi DML IRM	Unconfoundedness, Multi class Overlap, SUTVA, No leakage, Score stability
Synthetic Control	ASCM	No interference / spillovers, No anticipation, The treated unit’s untreated outcome path is well approximated by the donor pool
Difference in Difference	CallawaySantAnnaDID	Parallel trends, no anticipation, stable group composition, no spillovers between treated and control groups.
IV	DML IV	First-stage strength, Reduced form, Instrument balance by Z, Instrument propensity / predictability
Uplift / CATE scoring	DML IRM (CATE)	Identified treatment effects from randomized or unconfounded data, overlap, calibrated individual-level predictions.

Introduction to Causal Inference: guide

See scenario notebooks: https://causalis.causalcraft.com/explore-scenarios

Contributing guidelines

Maintainers

Ioann Martynov

References

https://github.com/DoubleML/doubleml-for-py

Search terms / supported methods

Causalis covers methods often searched as:

• causal inference Python
• causal machine learning Python
• treatment effect estimation
• A/B testing Python
• randomized controlled trial analysis
• CUPED Python
• Double Machine Learning Python
• DML / IRM
• CATE estimation
• uplift modeling
• propensity score diagnostics
• synthetic control Python
• difference-in-differences Python