Should this line be "del matched_treated['treated_index']"?

[ZJianL]

https://github.com/akelleh/causality/blob/aa458538224ec162b751c1e225b1ff02cb474191/causality/estimation/parametric.py#L201C34-L201C34
In the get_treated_matches function of the PropensityScoringModel class, after join, did you mean to delete the treated_index instead of control_index? Because I think what we need in the matched_treated is the control_index which has been deleted in this version.

[cjqjyy]

I think so too.

[cjqjyy]

https://github.com/akelleh/causality/blob/aa458538224ec162b751c1e225b1ff02cb474191/causality/estimation/parametric.py#L201C34-L201C34
del matched_treated['control_index'] should be del matched_treated['treated_index']