Skip to content

Hierarchical Causal Models #370

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
cthoyt merged 140 commits into from
Jan 15, 2026
Merged

Hierarchical Causal Models #370

cthoyt merged 140 commits into from
Jan 15, 2026

Conversation

@cthoyt
Copy link
Member

@cthoyt cthoyt commented Jan 15, 2026 •
edited
Loading

Closes #278

This PR splits out the code parts of #236 and leaves the notebooks for finishing later.

Adam Rupe and others added 30 commits February 27, 2025 18:43
@adamrupe
functions for creating and querying HCMs with pygraphviz
3c27650
@adamrupe
functionality to collapse phygraphviz HCMs to nxmixedgraphs
3975ba4
@adamrupe
added __all__
134202c
@adamrupe
added test_hierarchical
e643395
@adamrupe
pygraphviz dep and lint exception for hierarchical in pyproject
4079a0d
@adamrupe
auto lint hierarchical
e8fb669
@adamrupe
auto lint test_hierarchical
c76a6f9
@adamrupe
added docstrings to hierarchical.py
9134417
@adamrupe
added docstrings to test_hierarchical.py
833ed32
@adamrupe
more ignores in pyproject.toml
6828af3
@adamrupe
imperative docstrings
06796ea
@adamrupe
cleaned whitespace
f7b6da8
@adamrupe
lint HCM_from_lists
61fcfc1
@adamrupe
more linting
63070c0
@adamrupe
added typing to hierarchical.py
7fa1594
@adamrupe
added typing to test_hierarchical.py
f36ad10
@adamrupe
HCM notebook with Fig 2 up to collapsed models
0709527
@adamrupe
lint fixes
cfed46d
@adamrupe
docstring lint
5e132c9
@adamrupe
test fixtures for Fig 2 augmented models
7390162
@adamrupe
added augment_collapsed_model function that implements Algorithm 2
609cd42
@adamrupe
tox formatting
eb1bdda
@adamrupe
tests for augmentation and formatting
69d40ff
@adamrupe
added augmented models to notebook
1a8fdf3
@adamrupe
add marginalize function implementing Algorithm 3
64dbada
@adamrupe
Instrument test for algorithm 3
cce08cd
@cthoyt @adamrupe
Clean up literals
07e1ef6
@adamrupe
update collapse_HCM to preserve unit-level edges
dc6124b
@adamrupe
update augment and marginalize functions to return new objects instea…
57f975d 
…d of mutate input
@adamrupe
added copy_HCM
5613ddf
Nathaniel Merrill and others added 22 commits September 24, 2025 14:20
more small Narrative changes
539eb8d
Show Identifiability before technical discussion
a0e5c1b
Cleanup markdown code statements. Make descriptions less technical
8517e3b
@cthoyt
Update Counterfactual Transportability edited.ipynb
50b205f
@cthoyt
Merge branch 'main' into HCM-fig2
17e5ed8
Changed explanation of Collapse
17d1323
Resolved merge conflict in Hierarchical Causal Models.ipynb
edaa0c7
Rewrite the Augment section to be more readable
a9753a7
renamed variables to maintain consistency
7c2db57
Normalize line endings to LF
f41cfdb
Section title changes and small clarifying comments
e199c6f
@cthoyt
Merge branch 'main' into HCM-fig2
37cb36d
@cthoyt
Update Hierarchical Causal Models.ipynb
750b75f
@cthoyt
Update README.md
08a94dc
@cthoyt
Improve testing feedback
811e1c7
@djinnome
Fixed incorrect implementation of school confounder and interference …
db7b8cd 
…HSCM example
@djinnome
Made an alias to get_confounder_and_interference_h* from get_school_c…
5e24cf4 
…onfounder_and_interference_hc*
@cthoyt
Remove notebooks
f5ad111
@cthoyt
Update hierarchical.py
d6f621f
@cthoyt
Update tox.ini
b6d440d
@cthoyt
Update tox.ini
df7a297
@cthoyt
Update tests.yml
7df7a94
@cthoyt cthoyt marked this pull request as ready for review January 15, 2026 09:00
@cthoyt cthoyt changed the title HCM reduced Hierarchical Causal Models Jan 15, 2026
@cthoyt cthoyt enabled auto-merge (squash) January 15, 2026 09:08
@cthoyt cthoyt merged commit cc98c44 into main Jan 15, 2026
9 checks passed
@cthoyt cthoyt deleted the hcm-reduced branch January 15, 2026 09:11
@cthoyt cthoyt mentioned this pull request Jan 15, 2026
Merged
6 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

Hierarchical Causal Models

2 participants

@cthoyt @djinnome