moderndid.core.converters.doseresult_to_polars#

moderndid.core.converters.doseresult_to_polars(result: DoseResult, effect_type: str = 'att') → pl.DataFrame[source]#

Convert DoseResult to polars DataFrame for plotting.

Parameters:

resultDoseResult: Continuous treatment dose-response result.
effect_type{‘att’, ‘acrt’}, default=’att’: Type of effect to extract: - ‘att’: Average Treatment Effect on Treated - ‘acrt’: Average Causal Response on Treated

Returns:

polars.DataFrame

DataFrame with columns:

dose: dose level
effect: effect estimate (ATT or ACRT)
se: standard error
ci_lower: lower confidence interval
ci_upper: upper confidence interval

Raises:

ValueError: If effect_type is invalid or required data is missing.