Spaces:
Running
Running
MilesCranmer
commited on
Fix `julia_state_` typo
Browse files- pysr/param_groupings.yml +0 -1
- 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.
|
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.
|
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,
|