Spaces:
Running
Running
tttc3
commited on
Commit
•
e790ec3
1
Parent(s):
7ed402e
Hotfix for selection test
Browse files- 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_]
|