get ruff format pre commit working (hopefully)

Author: drbenvincent

causalpy/pymc_models.py

@@ -164,7 +164,7 @@ def print_row(
         ) -> None:
             """Print one row of the coefficient table"""
             formatted_name = f"  {name: <{max_label_length}}"
-            formatted_val = f"{round_num(coeff_samples.mean().data, round_to)}, 94% HDI [{round_num(coeff_samples.quantile(0.03).data, round_to)}, {round_num(coeff_samples.quantile(1-0.03).data, round_to)}]"  # noqa: E501
+            formatted_val = f"{round_num(coeff_samples.mean().data, round_to)}, 94% HDI [{round_num(coeff_samples.quantile(0.03).data, round_to)}, {round_num(coeff_samples.quantile(1 - 0.03).data, round_to)}]"  # noqa: E501
             print(f"  {formatted_name}  {formatted_val}")
 
         print("Model coefficients:")

causalpy/tests/test_api_stability.py

@@ -36,9 +36,9 @@ def test_causal_inference_and_discovery_with_python_example():
 
     # Build the model
     model = cp.pymc_models.WeightedSumFitter(sample_kwargs=sample_kwargs)
-    assert isinstance(
-        model, WeightedSumFitter
-    ), "model is not an instance of WeightedSumFitter"
+    assert isinstance(model, WeightedSumFitter), (
+        "model is not an instance of WeightedSumFitter"
+    )
 
     formula = "twitter ~ 0 + tiktok + linkedin + instagram"
 
@@ -49,9 +49,9 @@ def test_causal_inference_and_discovery_with_python_example():
         formula=formula,
         model=model,
     )
-    assert isinstance(
-        results, SyntheticControl
-    ), "results is not an instance of SyntheticControl"
+    assert isinstance(results, SyntheticControl), (
+        "results is not an instance of SyntheticControl"
+    )
 
     fig, ax = results.plot()
     assert isinstance(fig, plt.Figure)

docs/source/notebooks/multi_cell_geolift.ipynb

@@ -1099,7 +1099,7 @@
     "unpooled_results = []\n",
     "\n",
     "for i, target_geo in enumerate(treated):\n",
-    "    print(f\"Analyzing test geo: {target_geo} ({i+1} of {len(treated)})\")\n",
+    "    print(f\"Analyzing test geo: {target_geo} ({i + 1} of {len(treated)})\")\n",
     "    formula = f\"{target_geo} ~ 0 + {' + '.join(untreated)}\"\n",
     "    print(formula)\n",
     "\n",

docs/source/notebooks/rkink_pymc.ipynb

@@ -108,7 +108,7 @@
     "    beta = rng.random(5)\n",
     "    ax[0, col].plot(x, f(x, beta, kink), lw=3)\n",
     "    ax[1, col].plot(x, np.gradient(f(x, beta, kink), x), lw=3)\n",
-    "    ax[0, col].set(title=f\"Random  {col+1}\")\n",
+    "    ax[0, col].set(title=f\"Random  {col + 1}\")\n",
     "    ax[1, col].set(xlabel=\"x\")\n",
     "\n",
     "ax[0, 0].set(ylabel=\"$y = f(x)$\")\n",