Merge branch 'master' of github.com:DoubleML/doubleml-for-py into 0.3.X

DoubleML · Jun 4, 2021 · 1e44ef3 · 1e44ef3
2 parents 7402327 + 1e792bc
commit 1e44ef3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doubleml/double_ml_irm.py b/doubleml/double_ml_irm.py
@@ -192,7 +192,7 @@ def _score_elements(self, y, d, g_hat0, g_hat1, m_hat, smpls):
         # fraction of treated for ATTE
         p_hat = None
         if self.score == 'ATTE':
-            p_hat = np.zeros_like(d, dtype='float64')
+            p_hat = np.full_like(d, np.nan, dtype='float64')
             for _, test_index in smpls:
                 p_hat[test_index] = np.mean(d[test_index])