Spaces:
Running
Running
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:
|