MilesCranmer commited on
Commit
2621e9c
1 Parent(s): 2c97b85

Fix for operator assertion - can set EITHER unary or binary in csv loading

Browse files
Files changed (1) hide show
  1. 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.