deepsource-autofix[bot] commited on
Commit
781f479
1 Parent(s): 10bac39

Format code with black

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