Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
a4ab4f4
1
Parent(s):
5df9acb
Fix latex_table assertion for multi-output
Browse files- 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
|
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:
|