diff --git a/pysr/regressor_sequence.py b/pysr/regressor_sequence.py index d7eab8ad..aa17337f 100644 --- a/pysr/regressor_sequence.py +++ b/pysr/regressor_sequence.py @@ -131,7 +131,10 @@ def fit( self : object Fitted estimator. """ - X = self._validate_data(X) + X = self._validate_data(X, ensure_2d=False) + if X.ndim == 1: + X = X.reshape(-1, 1) + assert X.ndim == 2 _check_assertions( X, self.recursive_history_length, @@ -195,7 +198,10 @@ def predict(self, X, index=None, num_predictions=1): ValueError Raises if the `best_equation` cannot be evaluated. """ - X = self._validate_data(X) + X = self._validate_data(X, ensure_2d=False) + if X.ndim == 1: + X = X.reshape(-1, 1) + assert X.ndim == 2 _check_assertions(X, recursive_history_length=self.recursive_history_length) historical_X = self._sliding_window(X)[:: X.shape[1], :] if num_predictions < 1: