Spaces:

MilesCranmer
/

PySR

Running

MilesCranmer commited on Feb 5, 2023

Commit

c51eafe

•

2 Parent(s): 4068c18 a4ab4f4

Merge pull request #253 from MilesCranmer/MilesCranmer-patch-1

Files changed (1) hide show

pysr/sr.py CHANGED Viewed

@@ -2231,7 +2231,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
             if indices is not None:
                 assert isinstance(indices, list)
                 assert isinstance(indices[0], list)
-                assert isinstance(len(indices), self.nout_)
             generator_fnc = generate_multiple_tables
         else:

             if indices is not None:
                 assert isinstance(indices, list)
                 assert isinstance(indices[0], list)
+                assert len(indices) == self.nout_
             generator_fnc = generate_multiple_tables
         else: