diff --git a/ppi_py/ppi.py b/ppi_py/ppi.py index 9899f6a..ad078ea 100644 --- a/ppi_py/ppi.py +++ b/ppi_py/ppi.py @@ -52,13 +52,7 @@ def _rectified_p_value( def ppi_mean_pointestimate( - Y, - Yhat, - Yhat_unlabeled, - lhat=None, - coord=None, - w=None, - w_unlabeled=None + Y, Yhat, Yhat_unlabeled, lhat=None, coord=None, w=None, w_unlabeled=None ): """Computes the prediction-powered point estimate of the mean. @@ -926,6 +920,7 @@ def rectified_logistic_grad(_theta): else: return ppi_pointest + @njit def _logistic_get_stats( pointest,