plot_ATE -> plot_ate

Author: drbenvincent

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:

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)

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);"
             ]
         },
         {