all tests now passing 🎉 (one failing doctest)

Author: drbenvincent

causalpy/experiments/diff_in_diff.py

@@ -116,7 +116,7 @@ def __init__(
         self.y = xr.DataArray(
             self.y[:, 0],
             dims=["obs_ind"],
-            coords={"obs_ind": self.data.index},
+            coords={"obs_ind": np.arange(self.y.shape[0])},
         )
 
         # fit model
@@ -200,10 +200,10 @@ def __init__(
                     )
         elif isinstance(self.model, RegressorMixin):
             # This is the coefficient on the interaction term
-            # TODO: THIS IS NOT YET CORRECT ?????
+            # TODO: CHECK FOR CORRECTNESS
             self.causal_impact = (
                 self.y_pred_treatment[1] - self.y_pred_counterfactual[0]
-            )[0]
+            )
         else:
             raise ValueError("Model type not recognized")