MilesCranmer commited on
Commit
fa9d281
1 Parent(s): 75c23d4

docs: update __getstate__ docstring

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -9
pysr/sr.py CHANGED
@@ -1057,15 +1057,8 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1057
  Handle pickle serialization for PySRRegressor.
1058
 
1059
  The Scikit-learn standard requires estimators to be serializable via
1060
- `pickle.dumps()`. However, `PyCall.jlwrap` does not support pickle
1061
- serialization.
1062
-
1063
- Thus, for `PySRRegressor` to support pickle serialization, the
1064
- `julia_state_stream_` attribute must be hidden from pickle. This will
1065
- prevent the `warm_start` of any model that is loaded via `pickle.loads()`,
1066
- but does allow all other attributes of a fitted `PySRRegressor` estimator
1067
- to be serialized. Note: Jax and Torch format equations are also removed
1068
- from the pickled instance.
1069
  """
1070
  state = self.__dict__
1071
  show_pickle_warning = not (
 
1057
  Handle pickle serialization for PySRRegressor.
1058
 
1059
  The Scikit-learn standard requires estimators to be serializable via
1060
+ `pickle.dumps()`. However, some attributes do not support pickling
1061
+ and need to be hidden, such as the JAX and Torch representations.
 
 
 
 
 
 
 
1062
  """
1063
  state = self.__dict__
1064
  show_pickle_warning = not (