Skip to content

Commit

Permalink
fixed bug where CATE is not returned by lr
Browse files Browse the repository at this point in the history
Signed-off-by: Amit Sharma <amit_sharma@live.com>
  • Loading branch information
amit-sharma committed Dec 2, 2023
1 parent 918efc6 commit c5fea7e
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions dowhy/causal_estimators/regression_estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,8 @@ def estimate_effect(
self._target_units = target_units
self._treatment_value = treatment_value
self._control_value = control_value
if need_conditional_estimates is None:
need_conditional_estimates = self.need_conditional_estimates
# TODO make treatment_value and control value also as local parameters
# All treatments are set to the same constant value
effect_estimate = self._do(data, treatment_value) - self._do(data, control_value)
Expand Down

0 comments on commit c5fea7e

Please sign in to comment.