MilesCranmer commited on
Commit
83ef326
·
unverified ·
1 Parent(s): 5921e9c

Fix `julia_state_` typo

Browse files
Files changed (2) hide show
  1. pysr/param_groupings.yml +0 -1
  2. pysr/sr.py +4 -2
pysr/param_groupings.yml CHANGED
@@ -89,7 +89,6 @@
89
  - tempdir
90
  - delete_tempfiles
91
  - update
92
- - julia_kwargs
93
  - Exporting the Results:
94
  - equation_file
95
  - output_jax_format
 
89
  - tempdir
90
  - delete_tempfiles
91
  - update
 
92
  - Exporting the Results:
93
  - equation_file
94
  - output_jax_format
pysr/sr.py CHANGED
@@ -605,6 +605,8 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
605
  raw_julia_state_stream_ : ndarray
606
  The serialized state for the julia SymbolicRegression.jl backend (after fitting),
607
  stored as an array of uint8, produced by Julia's Serialization.serialize function.
 
 
608
  equation_file_contents_ : list[pandas.DataFrame]
609
  Contents of the equation file output by the Julia backend.
610
  show_pickle_warnings_ : bool
@@ -1131,7 +1133,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1131
  "for the raw stream of bytes.",
1132
  FutureWarning,
1133
  )
1134
- return self.julia_state
1135
 
1136
  def get_best(self, index=None):
1137
  """
@@ -1735,7 +1737,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1735
  options=options,
1736
  numprocs=cprocs,
1737
  parallelism=parallelism,
1738
- saved_state=self.julia_state,
1739
  return_state=True,
1740
  addprocs_function=cluster_manager,
1741
  heap_size_hint_in_bytes=self.heap_size_hint_in_bytes,
 
605
  raw_julia_state_stream_ : ndarray
606
  The serialized state for the julia SymbolicRegression.jl backend (after fitting),
607
  stored as an array of uint8, produced by Julia's Serialization.serialize function.
608
+ julia_state_ : ndarray
609
+ The deserialized state.
610
  equation_file_contents_ : list[pandas.DataFrame]
611
  Contents of the equation file output by the Julia backend.
612
  show_pickle_warnings_ : bool
 
1133
  "for the raw stream of bytes.",
1134
  FutureWarning,
1135
  )
1136
+ return self.julia_state_
1137
 
1138
  def get_best(self, index=None):
1139
  """
 
1737
  options=options,
1738
  numprocs=cprocs,
1739
  parallelism=parallelism,
1740
+ saved_state=self.julia_state_,
1741
  return_state=True,
1742
  addprocs_function=cluster_manager,
1743
  heap_size_hint_in_bytes=self.heap_size_hint_in_bytes,