Update double_ml_cvar.py

Author: SvenKlaassen

doubleml/double_ml_cvar.py

@@ -79,16 +79,16 @@ class DoubleMLCVAR(LinearScoreMixin, DoubleML):
     >>> import numpy as np
     >>> import doubleml as dml
     >>> from doubleml.datasets import make_irm_data
-    >>> from sklearn.ensemble import RandomForestClassifier
+    >>> from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
     >>> np.random.seed(3141)
-    >>> ml_g = RandomForestClassifier(n_estimators=100, max_features=20, max_depth=10, min_samples_leaf=2)
+    >>> ml_g = RandomForestRegressor(n_estimators=100, max_features=20, max_depth=10, min_samples_leaf=2)
     >>> ml_m = RandomForestClassifier(n_estimators=100, max_features=20, max_depth=10, min_samples_leaf=2)
     >>> data = make_irm_data(theta=0.5, n_obs=500, dim_x=20, return_type='DataFrame')
     >>> obj_dml_data = dml.DoubleMLData(data, 'y', 'd')
     >>> dml_cvar_obj = dml.DoubleMLCVAR(obj_dml_data, ml_g, ml_m, treatment=1, quantile=0.5)
     >>> dml_cvar_obj.fit().summary
-           coef   std err          t         P>|t|     2.5 %   97.5 %
-    d  1.462533  0.075714  19.316536  3.899567e-83  1.314136  1.61093
+           coef   std err          t         P>|t|     2.5 %    97.5 %
+    d  1.591441  0.095781  16.615498  5.382582e-62  1.403715  1.779167
     """
 
     def __init__(self,