formatting

Author: drbenvincent

causalpy/experiments/diff_in_diff.py

@@ -116,9 +116,7 @@ def __init__(
         self.y = xr.DataArray(
             self.y[:, 0],
             dims=["obs_ind"],
-            coords={
-                "obs_ind": self.data.index,
-            },
+            coords={"obs_ind": self.data.index},
         )
 
         # fit model

causalpy/experiments/interrupted_time_series.py

@@ -122,9 +122,7 @@ def __init__(
         self.pre_y = xr.DataArray(
             self.pre_y[:, 0],
             dims=["obs_ind"],
-            coords={
-                "obs_ind": self.datapre.index,
-            },
+            coords={"obs_ind": self.datapre.index},
         )
         self.post_X = xr.DataArray(
             self.post_X,
@@ -137,9 +135,7 @@ def __init__(
         self.post_y = xr.DataArray(
             self.post_y[:, 0],
             dims=["obs_ind"],
-            coords={
-                "obs_ind": self.datapost.index,
-            },
+            coords={"obs_ind": self.datapost.index},
         )
 
         # fit the model to the observed (pre-intervention) data