import numpy as np
from metacausal.estimators import CateEstimate, ComponentCateEstimate
from metacausal.plots import cate_profile

x = np.linspace(0, 1, 5)
result = CateEstimate(
    cate=x * 2,
    component_estimates={"a": ComponentCateEstimate(cate=x * 2 + 0.1)},
    aggregation="Median",
)
cate_profile(result, x, xlabel="x")