Spaces:
Running
Running
Fix for operator assertion - can set EITHER unary or binary in csv loading
Browse files- pysr/sr.py +1 -2
pysr/sr.py
CHANGED
|
@@ -1014,8 +1014,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 1014 |
f"{pkl_filename} does not exist, "
|
| 1015 |
"so we must create the model from scratch."
|
| 1016 |
)
|
| 1017 |
-
assert binary_operators is not None
|
| 1018 |
-
assert unary_operators is not None
|
| 1019 |
assert n_features_in is not None
|
| 1020 |
|
| 1021 |
# TODO: copy .bkup file if exists.
|
|
|
|
| 1014 |
f"{pkl_filename} does not exist, "
|
| 1015 |
"so we must create the model from scratch."
|
| 1016 |
)
|
| 1017 |
+
assert binary_operators is not None or unary_operators is not None
|
|
|
|
| 1018 |
assert n_features_in is not None
|
| 1019 |
|
| 1020 |
# TODO: copy .bkup file if exists.
|