Spaces:
Running
Running
MilesCranmer
commited on
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:
|