plot_ATE -> plot_ate

pymc-labs · Jul 11, 2024 · 47d4479 · 47d4479
1 parent 4b8141d
commit 47d4479
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/causalpy/experiments/inverse_propensity_weighting.py b/causalpy/experiments/inverse_propensity_weighting.py
@@ -238,7 +238,7 @@ def get_ate(self, i, idata, method="doubly_robust"):
         ate = trt - ntrt
         return [ate, trt, ntrt]
 
-    def plot_ATE(self, idata=None, method=None, prop_draws=100, ate_draws=300):
+    def plot_ate(self, idata=None, method=None, prop_draws=100, ate_draws=300):
         if idata is None:
             idata = self.model.idata
         if method is None:

diff --git a/causalpy/tests/test_misc.py b/causalpy/tests/test_misc.py
@@ -82,7 +82,7 @@ def test_inverse_prop():
     assert isinstance(ate_list, list)
     ate_list = result.get_ate(0, result.idata, method="overlap")
     assert isinstance(ate_list, list)
-    fig = result.plot_ATE(prop_draws=1, ate_draws=10)
+    fig = result.plot_ate(prop_draws=1, ate_draws=10)
     assert isinstance(fig, mpl.figure.Figure)
     fig = result.plot_balance_ecdf("age")
     assert isinstance(fig, mpl.figure.Figure)
diff --git a/docs/source/notebooks/inv_prop_pymc.ipynb b/docs/source/notebooks/inv_prop_pymc.ipynb
@@ -790,7 +790,7 @@
                 }
             ],
             "source": [
-                "result1.plot_ATE(method=\"raw\", prop_draws=10, ate_draws=500);"
+                "result1.plot_ate(method=\"raw\", prop_draws=10, ate_draws=500);"
             ]
         },
         {
@@ -825,7 +825,7 @@
                 }
             ],
             "source": [
-                "result1.plot_ATE(method=\"overlap\", prop_draws=10, ate_draws=500);"
+                "result1.plot_ate(method=\"overlap\", prop_draws=10, ate_draws=500);"
             ]
         },
         {
@@ -1108,7 +1108,7 @@
                 }
             ],
             "source": [
-                "result.plot_ATE(method=\"robust\", prop_draws=10, ate_draws=500);"
+                "result.plot_ate(method=\"robust\", prop_draws=10, ate_draws=500);"
             ]
         },
         {
@@ -1128,7 +1128,7 @@
                 }
             ],
             "source": [
-                "result.plot_ATE(method=\"doubly robust\", prop_draws=10, ate_draws=500);"
+                "result.plot_ate(method=\"doubly robust\", prop_draws=10, ate_draws=500);"
             ]
         },
         {