Spaces:
Running
Running
Commit
·
85371bb
1
Parent(s):
179fef6
Allow pickling without equations_ stored
Browse files- pysr/sr.py +3 -1
pysr/sr.py
CHANGED
|
@@ -883,7 +883,9 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 883 |
key: None if key == "raw_julia_state_" else value
|
| 884 |
for key, value in state.items()
|
| 885 |
}
|
| 886 |
-
if "equations_" in pickled_state
|
|
|
|
|
|
|
| 887 |
pickled_state["output_torch_format"] = False
|
| 888 |
pickled_state["output_jax_format"] = False
|
| 889 |
if self.nout_ == 1:
|
|
|
|
| 883 |
key: None if key == "raw_julia_state_" else value
|
| 884 |
for key, value in state.items()
|
| 885 |
}
|
| 886 |
+
if ("equations_" in pickled_state) and (
|
| 887 |
+
pickled_state["equations_"] is not None
|
| 888 |
+
):
|
| 889 |
pickled_state["output_torch_format"] = False
|
| 890 |
pickled_state["output_jax_format"] = False
|
| 891 |
if self.nout_ == 1:
|