Skip to content

Commit 16c1f1f

Browse files
fix(registry): wire conformal_continuous + conformal_interference
Add two missing registry entries so they surface in sp.list_functions() and sp.help() alongside the other P1.3 conformal extensions. Brings visible roadmap functions to 36/36 and total registered to 836. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
1 parent 05b0933 commit 16c1f1f

1 file changed

Lines changed: 41 additions & 0 deletions

File tree

src/statspai/registry.py

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2013,6 +2013,47 @@ def _build_registry():
20132013
reference="arXiv:2404.04794 (2024).",
20142014
))
20152015

2016+
# -- Continuous + interference conformal ------------------------ #
2017+
register(FunctionSpec(
2018+
name="conformal_continuous",
2019+
category="conformal_causal",
2020+
description=(
2021+
"Split-conformal prediction bands for continuous-treatment "
2022+
"dose-response curves (Kim, Jeong, Barber, Lee 2024)."
2023+
),
2024+
params=[
2025+
ParamSpec("data", "DataFrame", True),
2026+
ParamSpec("y", "str", True),
2027+
ParamSpec("treatment", "str", True),
2028+
ParamSpec("covariates", "list", True),
2029+
ParamSpec("test_data", "DataFrame", True),
2030+
ParamSpec("alpha", "float", False, 0.1),
2031+
],
2032+
returns="ContinuousConformalResult",
2033+
tags=["conformal", "continuous_treatment", "dose_response"],
2034+
reference="arXiv:2407.03094 (2024).",
2035+
))
2036+
register(FunctionSpec(
2037+
name="conformal_interference",
2038+
category="conformal_causal",
2039+
description=(
2040+
"Cluster-exchangeable split-conformal prediction under "
2041+
"network interference (2509.21660 systematic review)."
2042+
),
2043+
params=[
2044+
ParamSpec("data", "DataFrame", True),
2045+
ParamSpec("y", "str", True),
2046+
ParamSpec("treatment", "str", True),
2047+
ParamSpec("cluster", "str", True),
2048+
ParamSpec("covariates", "list", True),
2049+
ParamSpec("test_clusters", "list", True),
2050+
ParamSpec("alpha", "float", False, 0.1),
2051+
],
2052+
returns="InterferenceConformalResult",
2053+
tags=["conformal", "interference", "cluster"],
2054+
reference="arXiv:2509.21660 (2025).",
2055+
))
2056+
20162057
# -- Sharp OPE + Causal-Policy Forest ---------------------------- #
20172058
register(FunctionSpec(
20182059
name="sharp_ope_unobserved",

0 commit comments

Comments
 (0)