MilesCranmer commited on
Commit
a4ab4f4
1 Parent(s): 5df9acb

Fix latex_table assertion for multi-output

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -1
pysr/sr.py CHANGED
@@ -2231,7 +2231,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
2231
  if indices is not None:
2232
  assert isinstance(indices, list)
2233
  assert isinstance(indices[0], list)
2234
- assert isinstance(len(indices), self.nout_)
2235
 
2236
  generator_fnc = generate_multiple_tables
2237
  else:
 
2231
  if indices is not None:
2232
  assert isinstance(indices, list)
2233
  assert isinstance(indices[0], list)
2234
+ assert len(indices) == self.nout_
2235
 
2236
  generator_fnc = generate_multiple_tables
2237
  else: