We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a28c5da commit 642f651Copy full SHA for 642f651
causalpy/skl_models.py
@@ -16,7 +16,6 @@
16
from functools import partial
17
18
import numpy as np
19
-import xarray as xr
20
from scipy.optimize import fmin_slsqp
21
from sklearn.base import RegressorMixin
22
from sklearn.linear_model._base import LinearModel
@@ -29,9 +28,6 @@ class ScikitLearnAdaptor:
29
28
30
def calculate_impact(self, y_true, y_pred):
31
"""Calculate the causal impact of the intervention."""
32
- if isinstance(y_true, np.ndarray):
33
- y_true = xr.DataArray(y_true, dims=["obs_ind"])
34
-
35
return y_true - y_pred
36
37
def calculate_cumulative_impact(self, impact):
0 commit comments