resolve conflicts with main

Author: drbenvincent

.pre-commit-config.yaml

@@ -25,7 +25,7 @@ repos:
         exclude: &exclude_pattern 'iv_weak_instruments.ipynb'
         args: ["--maxkb=1500"]
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.11.10
+    rev: v0.11.13
     hooks:
       # Run the linter
       - id: ruff

CONTRIBUTING.md

@@ -167,20 +167,19 @@ We recommend that your contribution complies with the following guidelines befor
 
 ## Building the documentation locally
 
-A local build of the docs is achieved by:
+To build the documentation, run from the **project root**:
 
 ```bash
-cd docs
 make html
 ```
-
-Sometimes not all changes are recognised. In that case run this (again from within the `docs` folder):
-
+To clean and rebuild the documentation from scratch:
 ```bash
-make clean && make html
+make cleandocs
+make html
 ```
+ Docs are built in docs/_build/html, but these docs are not committed to the GitHub repository due to .gitignore.
 
-Docs are built in `docs/_build`, but these docs are _not_ committed to the GitHub repository due to `.gitignore`.
+ 📌 Note: The previous docs/Makefile has been removed. Please use only the root-level Makefile for documentation commands
 
 ## Overview of code structure
 

Makefile

@@ -1,4 +1,4 @@
-.PHONY: init lint check_lint test
+.PHONY: init lint check_lint test uml html cleandocs
 
 init:
 	python -m pip install -e . --no-deps
@@ -20,3 +20,12 @@ test:
 
 uml:
 	pyreverse -o png causalpy --output-directory docs/source/_static --ignore tests
+
+# Docs build commands
+
+html:
+	sphinx-build -b html docs/source docs/_build
+
+cleandocs:
+	rm -rf docs/_build
+	rm -rf docs/source/api/generated

README.md

@@ -4,7 +4,7 @@
 
 ----
 
-![Build Status](https://github.com/pymc-labs/CausalPy/workflows/ci/badge.svg?branch=main)
+![Build Status](https://github.com/pymc-labs/CausalPy/actions/workflows/ci.yml/badge.svg?branch=main)
 [![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 [![PyPI version](https://badge.fury.io/py/CausalPy.svg)](https://badge.fury.io/py/CausalPy)
 ![GitHub Repo stars](https://img.shields.io/github/stars/pymc-labs/causalpy?style=social)

causalpy/__init__.py

@@ -13,9 +13,7 @@
 #   limitations under the License.
 import arviz as az
 
-import causalpy.pymc_experiments as pymc_experiments  # to be deprecated
 import causalpy.pymc_models as pymc_models
-import causalpy.skl_experiments as skl_experiments  # to be deprecated
 import causalpy.skl_models as skl_models
 from causalpy.skl_models import create_causalpy_compatible_class
 from causalpy.version import __version__
@@ -43,11 +41,9 @@
     "InversePropensityWeighting",
     "load_data",
     "PrePostNEGD",
-    "pymc_experiments",  # to be deprecated
     "pymc_models",
     "RegressionDiscontinuity",
     "RegressionKink",
-    "skl_experiments",  # to be deprecated
     "skl_models",
     "SyntheticControl",
 ]

causalpy/experiments/diff_in_diff.py

@@ -19,6 +19,7 @@
 import numpy as np
 import pandas as pd
 import seaborn as sns
+import xarray as xr
 from matplotlib import pyplot as plt
 from patsy import build_design_matrices, dmatrices
 from sklearn.base import RegressorMixin
@@ -87,7 +88,8 @@ def __init__(
         **kwargs,
     ) -> None:
         super().__init__(model=model)
-
+        # rename the index to "obs_ind"
+        data.index.name = "obs_ind"
         self.data = data
         self.expt_type = "Difference in Differences"
         self.formula = formula
@@ -102,6 +104,21 @@ def __init__(
         self.y, self.X = np.asarray(y), np.asarray(X)
         self.outcome_variable_name = y.design_info.column_names[0]
 
+        # turn into xarray.DataArray's
+        self.X = xr.DataArray(
+            self.X,
+            dims=["obs_ind", "coeffs"],
+            coords={
+                "obs_ind": np.arange(self.X.shape[0]),
+                "coeffs": self.labels,
+            },
+        )
+        self.y = xr.DataArray(
+            self.y[:, 0],
+            dims=["obs_ind"],
+            coords={"obs_ind": np.arange(self.y.shape[0])},
+        )
+
         # fit model
         if isinstance(self.model, PyMCModel):
             COORDS = {"coeffs": self.labels, "obs_ind": np.arange(self.X.shape[0])}
@@ -183,13 +200,15 @@ 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")
 
+        return
+
     def input_validation(self):
         """Validate the input data and model formula for correctness"""
         if "post_treatment" not in self.formula: