Correct documentation

py-why · Feb 20, 2020 · c106485 · c106485
1 parent 51afc15
commit c106485
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/econml/dml.py b/econml/dml.py
@@ -652,7 +652,7 @@ def fit(self, Y, T, X=None, W=None, sample_weight=None, inference=None):
         inference: string, `Inference` instance, or None
             Method for performing inference.  This estimator supports 'bootstrap'
             (or an instance of :class:`.BootstrapInference`) and 'debiasedlasso'
-            (or an instance of :class:`.LinearCateInference`)
+            (or an instance of :class:`.LinearModelFinalInference`)
 
         Returns
         -------