Skip to content

Commit a17f5c9

Browse files
committed
fix some failing tests
1 parent 642f651 commit a17f5c9

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

causalpy/experiments/regression_discontinuity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def __init__(
133133
self.y = xr.DataArray(
134134
self.y[:, 0],
135135
dims=["obs_ind"],
136-
coords={"obs_ind": self.data.index},
136+
coords={"obs_ind": np.arange(self.y.shape[0])},
137137
)
138138

139139
# fit model

causalpy/experiments/regression_kink.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def __init__(
9696
self.y = xr.DataArray(
9797
self.y[:, 0],
9898
dims=["obs_ind"],
99-
coords={"obs_ind": self.data.index},
99+
coords={"obs_ind": np.arange(self.y.shape[0])},
100100
)
101101

102102
COORDS = {"coeffs": self.labels, "obs_ind": np.arange(self.X.shape[0])}

0 commit comments

Comments
 (0)