Spaces:
Running
Running
deepsource-autofix[bot]
commited on
Commit
•
781f479
1
Parent(s):
10bac39
Format code with black
Browse files- pysr/sr.py +3 -1
pysr/sr.py
CHANGED
@@ -274,7 +274,9 @@ class CallableEquation:
|
|
274 |
expected_shape = (X.shape[0],)
|
275 |
if isinstance(X, pd.DataFrame):
|
276 |
# Lambda function takes as argument:
|
277 |
-
return self._lambda(**{k: X[k].values for k in X.columns}) * np.ones(
|
|
|
|
|
278 |
elif self._selection is not None:
|
279 |
return self._lambda(*X[:, self._selection].T) * np.ones(expected_shape)
|
280 |
return self._lambda(*X.T) * np.ones(expected_shape)
|
|
|
274 |
expected_shape = (X.shape[0],)
|
275 |
if isinstance(X, pd.DataFrame):
|
276 |
# Lambda function takes as argument:
|
277 |
+
return self._lambda(**{k: X[k].values for k in X.columns}) * np.ones(
|
278 |
+
expected_shape
|
279 |
+
)
|
280 |
elif self._selection is not None:
|
281 |
return self._lambda(*X[:, self._selection].T) * np.ones(expected_shape)
|
282 |
return self._lambda(*X.T) * np.ones(expected_shape)
|