MilesCranmer commited on
Commit
688c82c
1 Parent(s): ae96818

Fix old docstring

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -2
pysr/sr.py CHANGED
@@ -536,8 +536,8 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
536
  :type precision: int
537
  :param **kwargs: Other options passed to SymbolicRegression.Options, for example, if you modify SymbolicRegression.jl to include additional arguments.
538
  :type **kwargs: dict
539
- :returns: Results dataframe, giving complexity, MSE, and equations (as strings), as well as functional forms. If list, each element corresponds to a dataframe of equations for each output.
540
- :type: pd.DataFrame/list
541
  """
542
  super().__init__()
543
  self.model_selection = model_selection
 
536
  :type precision: int
537
  :param **kwargs: Other options passed to SymbolicRegression.Options, for example, if you modify SymbolicRegression.jl to include additional arguments.
538
  :type **kwargs: dict
539
+ :returns: Initialized model. Call `.fit(X, y)` to fit your data!
540
+ :type: PySRRegressor
541
  """
542
  super().__init__()
543
  self.model_selection = model_selection