Spaces:
Running
Running
Commit
·
be72c13
1
Parent(s):
de4d559
More assertions for latex table export
Browse files- pysr/sr.py +4 -0
pysr/sr.py
CHANGED
|
@@ -2032,6 +2032,10 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 2032 |
|
| 2033 |
generator_fnc = generate_multiple_tables
|
| 2034 |
else:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2035 |
generator_fnc = generate_single_table
|
| 2036 |
|
| 2037 |
return generator_fnc(
|
|
|
|
| 2032 |
|
| 2033 |
generator_fnc = generate_multiple_tables
|
| 2034 |
else:
|
| 2035 |
+
if indices is not None:
|
| 2036 |
+
assert isinstance(indices, list)
|
| 2037 |
+
assert isinstance(indices[0], int)
|
| 2038 |
+
|
| 2039 |
generator_fnc = generate_single_table
|
| 2040 |
|
| 2041 |
return generator_fnc(
|