reset formatting

DoubleML · Dec 8, 2023 · d3cdedf · d3cdedf
1 parent 2de5b29
commit d3cdedf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/doubleml/double_ml_plr.py b/doubleml/double_ml_plr.py
@@ -150,7 +150,6 @@ def __init__(self,
 
         self._initialize_ml_nuisance_params()
         self._sensitivity_implemented = True
-
         self._external_predictions_implemented = True
 
     def _initialize_ml_nuisance_params(self):
@@ -173,7 +172,7 @@ def _nuisance_est(self, smpls, n_jobs_cv, external_predictions, return_models=Fa
                          force_all_finite=False)
         x, d = check_X_y(x, self._dml_data.d,
                          force_all_finite=False)
-        
+
         m_external = external_predictions['ml_m'] is not None
         l_external = external_predictions['ml_l'] is not None
         if 'ml_g' in self._learner:
@@ -224,8 +223,8 @@ def _nuisance_est(self, smpls, n_jobs_cv, external_predictions, return_models=Fa
             # nuisance g
             if g_external:
                 g_hat = {'preds': external_predictions['ml_g'],
-                        'targets': None,
-                        'models': None}
+                         'targets': None,
+                         'models': None}
             else:
                 g_hat = _dml_cv_predict(self._learner['ml_g'], x, y - theta_initial*d, smpls=smpls, n_jobs=n_jobs_cv,
                                         est_params=self._get_params('ml_g'), method=self._predict_method['ml_g'],