diff --git a/ngboost/ngboost.py b/ngboost/ngboost.py index c8aa998..d6666ac 100644 --- a/ngboost/ngboost.py +++ b/ngboost/ngboost.py @@ -338,7 +338,7 @@ def partial_fit( raise ValueError("y cannot be None") X, Y = check_X_y( - X, Y, accept_sparse=True, y_numeric=True, multi_output=self.multi_output + X, Y, accept_sparse=True, y_numeric=True, multi_output=self.multi_output, force_all_finite='allow-nan' ) self.n_features = X.shape[1]