Fix RScorer circular import

py-why · Apr 1, 2022 · ca6088e · ca6088e
1 parent 289c541
commit ca6088e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/econml/dml/causal_forest.py b/econml/dml/causal_forest.py
@@ -22,7 +22,6 @@
 from .._cate_estimator import LinearCateEstimator
 from .._shap import _shap_explain_multitask_model_cate
 from .._ortho_learner import _OrthoLearner
-from ..score import RScorer
 
 
 class _CausalForestFinalWrapper:
@@ -511,6 +510,8 @@ def __init__(self, *,
                  n_jobs=-1,
                  random_state=None,
                  verbose=0):
+        from ..score import RScorer
+
 
         # TODO: consider whether we need more care around stateful featurizers,
         #       since we clone it and fit separate copies