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

for more information, see https://pre-commit.ci
covid-19-impact-lab · Jan 31, 2022 · c53b805 · c53b805
1 parent d7bcbda
commit c53b805
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/setup.cfg b/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

diff --git a/src/sid/msm.py b/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,

diff --git a/src/sid/update_states.py b/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
     )