Add click install to fix ci (#193)

* Add black

Author: canyon289

.pre-commit-config.yaml

@@ -8,7 +8,7 @@ repos:
   rev: 0.5.6
   hooks:
     - id: nbqa-black
-      additional_dependencies: [black==20.8b1]
+      additional_dependencies: [black==22.3.0]
       files: ^(Rethinking_2|BSM)/
     - id: nbqa-isort
       additional_dependencies: [isort==5.6.4]

BSM/Chapter_03_00_Bayesian_CLT.ipynb

@@ -54,7 +54,7 @@
     "    A = Y - 0.5\n",
     "    B = n - Y - 0.5\n",
     "    θ_MAP = A / (A + B)\n",
-    "    info = A / θ_MAP ** 2 + B / (1 - θ_MAP) ** 2\n",
+    "    info = A / θ_MAP**2 + B / (1 - θ_MAP) ** 2\n",
     "\n",
     "    post1 = stats.binom(n, θ).pmf(Y) * stats.beta(0.5, 0.5).pdf(θ)\n",
     "    post1 = post1 / np.sum(post1)\n",

BSM/Chapter_03_01_Gibbs_sampling_one_sample_t-test.ipynb

@@ -142,7 +142,7 @@
     "    # sample mu|s2,Y\n",
     "    MN = np.sum(Y) / (n + m)\n",
     "    VR = s2 / (n + m)\n",
-    "    mu = stats.norm(MN, VR ** 0.5).rvs(1)\n",
+    "    mu = stats.norm(MN, VR**0.5).rvs(1)\n",
     "\n",
     "    # sample s2|mu,Y\n",
     "    A = a + n / 2\n",
@@ -291,7 +291,7 @@
     }
    ],
    "source": [
-    "keep_s = keep_s2 ** 0.5\n",
+    "keep_s = keep_s2**0.5\n",
     "plt.hist(keep_s2)\n",
     "plt.xlabel(\"sigma\")\n",
     "plt.title(\"Marginal posterior\");"

BSM/Chapter_03_02_Gibbs_sampling_two_sample_t-test.ipynb

@@ -122,12 +122,12 @@
     "    # sample muY|muZ,s2,Y,Z\n",
     "    A = np.sum(Y) / s2 + mu_0 / s2_0\n",
     "    B = n / s2 + 1 / s2_0\n",
-    "    muY = stats.norm(A / B, 1 / B ** 0.5).rvs(1)[0]\n",
+    "    muY = stats.norm(A / B, 1 / B**0.5).rvs(1)[0]\n",
     "\n",
     "    # sample muZ|muY,s2,Y,Z\n",
     "    A = np.sum(Z) / s2 + mu_0 / s2_0\n",
     "    B = m / s2 + 1 / s2_0\n",
-    "    muZ = stats.norm(A / B, 1 / B ** 0.5).rvs(1)[0]\n",
+    "    muZ = stats.norm(A / B, 1 / B**0.5).rvs(1)[0]\n",
     "\n",
     "    # sample s2|muY,muZ,Y,Z\n",
     "    A = n / 2 + m / 2 + a\n",

BSM/Chapter_03_03_Gibbs_sampling_for_simple_linear_regression.ipynb

@@ -394,12 +394,12 @@
     "    # sample alpha\n",
     "    V = n / s2 + mu_0 / s2_0\n",
     "    M = np.sum(Y - X * β) / s2 + 1 / s2_0\n",
-    "    α = stats.norm(M / V, 1 / V ** 0.5).rvs(1)[0]\n",
+    "    α = stats.norm(M / V, 1 / V**0.5).rvs(1)[0]\n",
     "\n",
     "    # sample beta\n",
-    "    V = np.sum(X ** 2) / s2 + mu_0 / s2_0\n",
+    "    V = np.sum(X**2) / s2 + mu_0 / s2_0\n",
     "    M = np.sum(X * (Y - α)) / s2 + 1 / s2_0\n",
-    "    β = stats.norm(M / V, 1 / V ** 0.5).rvs(1)[0]\n",
+    "    β = stats.norm(M / V, 1 / V**0.5).rvs(1)[0]\n",
     "\n",
     "    # sample s2|mu,Y,Z\n",
     "    A = n / 2 + a\n",

BSM/Chapter_03_09_Simple_linear_regression_in_PyMC3.ipynb

@@ -79,7 +79,7 @@
     "with pm.Model() as model:\n",
     "    # Priors\n",
     "    τ = pm.Gamma(\"τ\", 0.1, 10)\n",
-    "    σ = pm.Deterministic(\"σ\", 1 / (τ ** 0.5))\n",
+    "    σ = pm.Deterministic(\"σ\", 1 / (τ**0.5))\n",
     "    # σ = pm.HalfNormal('σ', np.std(mass))\n",
     "    β1 = pm.Normal(\"β1\", 0, 1000)\n",
     "    β2 = pm.Normal(\"β2\", 0, 1000)\n",

Rethinking_2/Chp_04.ipynb

@@ -3062,7 +3062,7 @@
    ],
    "source": [
     "d[\"weight_std\"] = (d.weight - d.weight.mean()) / d.weight.std()\n",
-    "d[\"weight_std2\"] = d.weight_std ** 2\n",
+    "d[\"weight_std2\"] = d.weight_std**2\n",
     "\n",
     "with pm.Model() as m_4_5:\n",
     "    a = pm.Normal(\"a\", mu=178, sd=100)\n",
@@ -3329,7 +3329,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "weight_m = np.vstack((d.weight_std, d.weight_std ** 2, d.weight_std ** 3))"
+    "weight_m = np.vstack((d.weight_std, d.weight_std**2, d.weight_std**3))"
    ]
   },
   {

Rethinking_2/Chp_06.ipynb

@@ -1217,7 +1217,7 @@
     "\n",
     "\n",
     "def sim_coll(r=0.9):\n",
-    "    x = np.random.normal(loc=r * d[\"perc.fat\"], scale=np.sqrt((1 - r ** 2) * np.var(d[\"perc.fat\"])))\n",
+    "    x = np.random.normal(loc=r * d[\"perc.fat\"], scale=np.sqrt((1 - r**2) * np.var(d[\"perc.fat\"])))\n",
     "    _, cov = curve_fit(mv, (d[\"perc.fat\"], x), d[\"kcal.per.g\"])\n",
     "    return np.sqrt(np.diag(cov))[-1]\n",
     "\n",

Rethinking_2/Chp_09.ipynb

@@ -139,7 +139,7 @@
    ],
    "source": [
     "def rad_dist(Y):\n",
-    "    return np.sqrt(np.sum(Y ** 2))\n",
+    "    return np.sqrt(np.sum(Y**2))\n",
     "\n",
     "\n",
     "fig, ax = plt.subplots(1, 1, figsize=[7, 3])\n",
@@ -207,8 +207,8 @@
     "def calc_U_gradient(x, y, q, a=0, b=1, k=0, d=1):\n",
     "    muy, mux = q\n",
     "\n",
-    "    G1 = np.sum(y - muy) + (a - muy) / b ** 2  # dU/dmuy\n",
-    "    G2 = np.sum(x - mux) + (k - mux) / b ** 2  # dU/dmux\n",
+    "    G1 = np.sum(y - muy) + (a - muy) / b**2  # dU/dmuy\n",
+    "    G2 = np.sum(x - mux) + (k - mux) / b**2  # dU/dmux\n",
     "\n",
     "    return np.array([-G1, -G2])"
    ]
@@ -257,9 +257,9 @@
     "    p *= -1\n",
     "    # Evaluate potential and kinetic energies sat start and end of trajectory\n",
     "    current_U = U(x, y, current_q)\n",
-    "    current_K = np.sum(current_p ** 2) / 2\n",
+    "    current_K = np.sum(current_p**2) / 2\n",
     "    proposed_U = U(x, y, q)\n",
-    "    proposed_K = np.sum(p ** 2) / 2\n",
+    "    proposed_K = np.sum(p**2) / 2\n",
     "    # Accept or reject the state at end of trajectory, returning either\n",
     "    # the position at the end of the trajectory or the initial position\n",
     "    accept = False\n",

Rethinking_2/Chp_10.ipynb

@@ -233,7 +233,7 @@
    ],
    "source": [
     "p = 0.7\n",
-    "A = [(1 - p) ** 2, p * (1 - p), (1 - p) * p, p ** 2]\n",
+    "A = [(1 - p) ** 2, p * (1 - p), (1 - p) * p, p**2]\n",
     "A"
    ]
   },
@@ -299,9 +299,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "H = np.zeros(10 ** 5)\n",
-    "p = np.zeros((10 ** 5, 4))\n",
-    "for rep in range(10 ** 5):\n",
+    "H = np.zeros(10**5)\n",
+    "p = np.zeros((10**5, 4))\n",
+    "for rep in range(10**5):\n",
     "    h, p_ = sim_p()\n",
     "    H[rep] = h\n",
     "    p[rep] = p_"