MilesCranmer commited on
Commit
be72c13
1 Parent(s): de4d559

More assertions for latex table export

Browse files
Files changed (1) hide show
  1. 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(