Fix linting issue

py-why · Jan 6, 2020 · 2a4d6db · 2a4d6db
1 parent a7bcdf6
commit 2a4d6db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/econml/drlearner.py b/econml/drlearner.py
@@ -328,7 +328,8 @@ def score(self, Y, T, X=None, W=None, *, nuisances, sample_weight=None, sample_v
  return np.mean(np.average((Y_pred[..., 1:] - Y_pred[..., [0]] - self.model_cate.predict(X))**2,
  weights=sample_weight, axis=0))
  else:
- return np.mean([np.average((Y_pred[..., t] - Y_pred[..., 0] - self.models_cate[t - 1].predict(X))**2,
+ return np.mean([np.average((Y_pred[..., t] - Y_pred[..., 0] -
+ self.models_cate[t - 1].predict(X))**2,
  weights=sample_weight, axis=0)
  for t in np.arange(1, Y_pred.shape[-1])])