tttc3 commited on
Commit
e790ec3
1 Parent(s): 7ed402e

Hotfix for selection test

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -1
pysr/sr.py CHANGED
@@ -1497,7 +1497,7 @@ class PySRRegressor(BaseEstimator, RegressorMixin, MultiOutputMixin):
1497
  X = pd.DataFrame(X)
1498
 
1499
  if isinstance(X.columns, pd.RangeIndex):
1500
- if self.selection_mask_:
1501
  # RangeIndex enforces column order allowing columns to
1502
  # be correctly filtered with self.selection_mask_
1503
  X = X.iloc[:, self.selection_mask_]
 
1497
  X = pd.DataFrame(X)
1498
 
1499
  if isinstance(X.columns, pd.RangeIndex):
1500
+ if self.selection_mask_ is not None:
1501
  # RangeIndex enforces column order allowing columns to
1502
  # be correctly filtered with self.selection_mask_
1503
  X = X.iloc[:, self.selection_mask_]