Spaces:
Running
Running
MilesCranmer
commited on
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(
|