[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

docs/source/_static/images/make_infectiousness_period.ipynb

@@ -126,7 +126,7 @@
    "source": [
     "dist_mean = 2 * 0.12 + 4 * 0.29 + 6 * 0.47 + 10 * 0.12\n",
     "dist_var = (\n",
-    "    2 ** 2 * 0.12 + 4 ** 2 * 0.29 + 6 ** 2 * 0.47 + 10 ** 2 * 0.12 - dist_mean ** 2\n",
+    "    2**2 * 0.12 + 4**2 * 0.29 + 6**2 * 0.47 + 10**2 * 0.12 - dist_mean**2\n",
     ")\n",
     "dist_std = np.sqrt(dist_var)\n",
     "dist_mean, dist_std"

docs/source/_static/images/make_time_until_death.ipynb

@@ -41,7 +41,7 @@
    "source": [
     "mean = 11.74\n",
     "std = 8.79\n",
-    "var = 8.79 ** 2\n",
+    "var = 8.79**2\n",
     "\n",
     "print(mean, std)"
    ]

docs/source/_static/images/make_time_until_icu_recovery.ipynb

@@ -41,7 +41,7 @@
    "source": [
     "mean = 18.8\n",
     "std = 12.21\n",
-    "var = std ** 2\n",
+    "var = std**2\n",
     "\n",
     "print(mean, std)"
    ]

setup.cfg

@@ -20,6 +20,7 @@ classifiers =
     Programming Language :: Python :: 3.7
     Programming Language :: Python :: 3.8
     Programming Language :: Python :: 3.9
+    Programming Language :: Python :: 3.10
     Topic :: Scientific/Engineering
 project_urls =
     Changelog = https://sid-dev.readthedocs.io/en/latest/changes.html

src/sid/msm.py

@@ -156,7 +156,7 @@ def _msm(
     moment_errors = flat_simulated_moments - flat_empirical_moments
 
     root_contribs = np.sqrt(np.diagonal(weighting_matrix)) * moment_errors
-    value = np.sum(root_contribs ** 2)
+    value = np.sum(root_contribs**2)
 
     out = {
         "value": value,

src/sid/update_states.py

@@ -289,7 +289,7 @@ def _get_waning_immunity_coefficients(
         ("immunity", "immunity_waning", f"slope_after_maximum_{event}"), "value"
     ]
 
-    slope_before_maximum = maximum_immunity / (time_to_reach_maximum ** 3)
+    slope_before_maximum = maximum_immunity / (time_to_reach_maximum**3)
     intercept_after_maximum = (
         maximum_immunity - slope_after_maximum * time_to_reach_maximum
     )