Update pysr/regressor_sequence.py

MilesCranmer · Aug 26, 2024 · 63e7785 · 63e7785
1 parent 66542ff
commit 63e7785
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/regressor_sequence.py b/pysr/regressor_sequence.py
@@ -101,7 +101,7 @@ def fit(
         Parameters
         ----------
         X : ndarray | pandas.DataFrame
-            Sequence of shape (n_times, n_features).
+            Sequence of shape (n_times, n_features) or (n_times,)
         weights : ndarray | pandas.DataFrame
             Weight array of the same shape as `X`.
             Each element is how to weight the mean-square-error loss