Skip to content

Commit a932fbe

Browse files
drbenvincentdrbenvincent
authored
Merge pull request #89 from pymc-labs/arviz-style-auto-applied
remove need for user to import arviz and set style
2 parents bc3697e + 5fb6b29 commit a932fbe

12 files changed

+190
-302
lines changed

‎causalpy/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
1+
import arviz as az
2+
13
import causalpy.pymc_experiments
24
import causalpy.pymc_models
35
import causalpy.skl_experiments
46
import causalpy.skl_models
57

68
from .data import load_data
9+
10+
az.style.use("arviz-darkgrid")

‎docs/notebooks/did_pymc.ipynb

Lines changed: 13 additions & 21 deletions
Large diffs are not rendered by default.

‎docs/notebooks/did_pymc_banks.ipynb

Lines changed: 22 additions & 44 deletions
Large diffs are not rendered by default.

‎docs/notebooks/did_skl.ipynb

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -13,31 +13,21 @@
1313
"metadata": {},
1414
"outputs": [],
1515
"source": [
16-
"import causalpy as cp\n",
17-
"import arviz as az"
16+
"import causalpy as cp"
1817
]
1918
},
2019
{
2120
"cell_type": "code",
2221
"execution_count": 2,
2322
"metadata": {},
2423
"outputs": [],
25-
"source": [
26-
"az.style.use(\"arviz-darkgrid\")"
27-
]
28-
},
29-
{
30-
"cell_type": "code",
31-
"execution_count": 3,
32-
"metadata": {},
33-
"outputs": [],
3424
"source": [
3525
"data = cp.load_data(\"did\")"
3626
]
3727
},
3828
{
3929
"cell_type": "code",
40-
"execution_count": 4,
30+
"execution_count": 3,
4131
"metadata": {},
4232
"outputs": [],
4333
"source": [
@@ -54,7 +44,7 @@
5444
},
5545
{
5646
"cell_type": "code",
57-
"execution_count": 5,
47+
"execution_count": 4,
5848
"metadata": {},
5949
"outputs": [
6050
{

‎docs/notebooks/rd_pymc.ipynb

Lines changed: 12 additions & 22 deletions
Large diffs are not rendered by default.

‎docs/notebooks/rd_pymc_drinking.ipynb

Lines changed: 27 additions & 37 deletions
Large diffs are not rendered by default.

‎docs/notebooks/rd_skl.ipynb

Lines changed: 10 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -13,19 +13,9 @@
1313
"metadata": {},
1414
"outputs": [],
1515
"source": [
16-
"import arviz as az\n",
1716
"import causalpy as cp"
1817
]
1918
},
20-
{
21-
"cell_type": "code",
22-
"execution_count": 2,
23-
"metadata": {},
24-
"outputs": [],
25-
"source": [
26-
"az.style.use(\"arviz-darkgrid\")"
27-
]
28-
},
2919
{
3020
"cell_type": "markdown",
3121
"metadata": {},
@@ -35,7 +25,7 @@
3525
},
3626
{
3727
"cell_type": "code",
38-
"execution_count": 3,
28+
"execution_count": 2,
3929
"metadata": {},
4030
"outputs": [
4131
{
@@ -108,7 +98,7 @@
10898
"4 -0.882469 -0.420811 False"
10999
]
110100
},
111-
"execution_count": 3,
101+
"execution_count": 2,
112102
"metadata": {},
113103
"output_type": "execute_result"
114104
}
@@ -127,7 +117,7 @@
127117
},
128118
{
129119
"cell_type": "code",
130-
"execution_count": 4,
120+
"execution_count": 3,
131121
"metadata": {},
132122
"outputs": [],
133123
"source": [
@@ -143,7 +133,7 @@
143133
},
144134
{
145135
"cell_type": "code",
146-
"execution_count": 5,
136+
"execution_count": 4,
147137
"metadata": {},
148138
"outputs": [
149139
{
@@ -170,7 +160,7 @@
170160
},
171161
{
172162
"cell_type": "code",
173-
"execution_count": 6,
163+
"execution_count": 5,
174164
"metadata": {},
175165
"outputs": [],
176166
"source": [
@@ -184,7 +174,7 @@
184174
},
185175
{
186176
"cell_type": "code",
187-
"execution_count": 7,
177+
"execution_count": 6,
188178
"metadata": {},
189179
"outputs": [
190180
{
@@ -211,7 +201,7 @@
211201
},
212202
{
213203
"cell_type": "code",
214-
"execution_count": 8,
204+
"execution_count": 7,
215205
"metadata": {},
216206
"outputs": [],
217207
"source": [
@@ -229,7 +219,7 @@
229219
},
230220
{
231221
"cell_type": "code",
232-
"execution_count": 9,
222+
"execution_count": 8,
233223
"metadata": {},
234224
"outputs": [
235225
{
@@ -256,7 +246,7 @@
256246
},
257247
{
258248
"cell_type": "code",
259-
"execution_count": 10,
249+
"execution_count": 9,
260250
"metadata": {},
261251
"outputs": [],
262252
"source": [
@@ -273,7 +263,7 @@
273263
},
274264
{
275265
"cell_type": "code",
276-
"execution_count": 11,
266+
"execution_count": 10,
277267
"metadata": {},
278268
"outputs": [
279269
{

‎docs/notebooks/rd_skl_drinking.ipynb

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
"metadata": {},
1818
"outputs": [],
1919
"source": [
20-
"import arviz as az\n",
2120
"import causalpy as cp"
2221
]
2322
},
@@ -26,15 +25,6 @@
2625
"execution_count": 2,
2726
"metadata": {},
2827
"outputs": [],
29-
"source": [
30-
"az.style.use(\"arviz-darkgrid\")"
31-
]
32-
},
33-
{
34-
"cell_type": "code",
35-
"execution_count": 3,
36-
"metadata": {},
37-
"outputs": [],
3828
"source": [
3929
"df = (\n",
4030
" cp.load_data(\"drinking\")\n",
@@ -46,7 +36,7 @@
4636
},
4737
{
4838
"cell_type": "code",
49-
"execution_count": 4,
39+
"execution_count": 3,
5040
"metadata": {},
5141
"outputs": [],
5242
"source": [
@@ -70,7 +60,7 @@
7060
},
7161
{
7262
"cell_type": "code",
73-
"execution_count": 5,
63+
"execution_count": 4,
7464
"metadata": {},
7565
"outputs": [
7666
{
@@ -90,7 +80,7 @@
9080
},
9181
{
9282
"cell_type": "code",
93-
"execution_count": 6,
83+
"execution_count": 5,
9484
"metadata": {},
9585
"outputs": [
9686
{

‎docs/notebooks/sc2_pymc.ipynb

Lines changed: 43 additions & 51 deletions
Large diffs are not rendered by default.

‎docs/notebooks/sc2_skl.ipynb

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -15,17 +15,7 @@
1515
"metadata": {},
1616
"outputs": [],
1717
"source": [
18-
"import causalpy as cp\n",
19-
"import arviz as az"
20-
]
21-
},
22-
{
23-
"cell_type": "code",
24-
"execution_count": 2,
25-
"metadata": {},
26-
"outputs": [],
27-
"source": [
28-
"az.style.use(\"arviz-darkgrid\")"
18+
"import causalpy as cp"
2919
]
3020
},
3121
{
@@ -37,7 +27,7 @@
3727
},
3828
{
3929
"cell_type": "code",
40-
"execution_count": 3,
30+
"execution_count": 2,
4131
"metadata": {},
4232
"outputs": [
4333
{
@@ -124,7 +114,7 @@
124114
"2010-05-31 5 2010 4 40.753973"
125115
]
126116
},
127-
"execution_count": 3,
117+
"execution_count": 2,
128118
"metadata": {},
129119
"output_type": "execute_result"
130120
}
@@ -149,7 +139,7 @@
149139
},
150140
{
151141
"cell_type": "code",
152-
"execution_count": 4,
142+
"execution_count": 3,
153143
"metadata": {},
154144
"outputs": [],
155145
"source": [
@@ -172,7 +162,7 @@
172162
},
173163
{
174164
"cell_type": "code",
175-
"execution_count": 5,
165+
"execution_count": 4,
176166
"metadata": {},
177167
"outputs": [
178168
{

0 commit comments

Comments
 (0)