remove the unused import and variable in sensitivity.py

uber · Apr 12, 2024 · 0f5db04 · 0f5db04
1 parent 95e5d4b
commit 0f5db04
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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