precommit

Author: mjhajharia

examples/case_studies/hierarchical_partial_pooling.ipynb

@@ -55,12 +55,12 @@
     }
    ],
    "source": [
+    "import aesara.tensor as at\n",
     "import arviz as az\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import pandas as pd\n",
     "import pymc3 as pm\n",
-    "import aesara.tensor as at\n",
     "\n",
     "%matplotlib inline"
    ]
@@ -127,7 +127,7 @@
     "N = len(hits)\n",
     "player_names = data.apply(lambda x: x.FirstName + \" \" + x.LastName, axis=1)\n",
     "coords = {\"player_names\": player_names.tolist()}\n",
-    "    \n",
+    "\n",
     "with pm.Model(coords=coords) as baseball_model:\n",
     "\n",
     "    phi = pm.Uniform(\"phi\", lower=0.0, upper=1.0)\n",
@@ -223,7 +223,7 @@
    ],
    "source": [
     "with baseball_model:\n",
-    "    trace = pm.sample(2000, tune=2000,chains=2, target_accept=0.95, return_inferencedata=True)\n",
+    "    trace = pm.sample(2000, tune=2000, chains=2, target_accept=0.95, return_inferencedata=True)\n",
     "\n",
     "    # check convergence diagnostics\n",
     "    assert all(az.rhat(trace) < 1.03)"
@@ -292,7 +292,7 @@
     }
    ],
    "source": [
-    "az.plot_forest(trace,var_names=[\"thetas\"]);"
+    "az.plot_forest(trace, var_names=[\"thetas\"]);"
    ]
   },
   {