diff --git a/causalml/metrics/sensitivity.py b/causalml/metrics/sensitivity.py index 8a7686ed..9462445e 100644 --- a/causalml/metrics/sensitivity.py +++ b/causalml/metrics/sensitivity.py @@ -144,9 +144,6 @@ def get_ate_ci(self, X, p, treatment, y): (numpy.ndarray): Mean and confidence interval (LB, UB) of the ATE estimate. """ - learner = self.learner - from ..inference.meta.tlearner import BaseTLearner - try: ate, ate_lower, ate_upper = self.learner.estimate_ate( X=X, p=p, treatment=treatment, y=y, return_ci=True