From f8ad0ed64685dfdea7848ca387c95d760296ba73 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 5 Dec 2024 16:07:11 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- mondey_backend/src/mondey_backend/routers/statistics.py | 8 ++++---- mondey_backend/tests/utils/test_statistics.py | 3 ++- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/mondey_backend/src/mondey_backend/routers/statistics.py b/mondey_backend/src/mondey_backend/routers/statistics.py index a68f3384..fba6eca1 100644 --- a/mondey_backend/src/mondey_backend/routers/statistics.py +++ b/mondey_backend/src/mondey_backend/routers/statistics.py @@ -51,10 +51,10 @@ def _finalize_statistics( return count, mean, np.sqrt(var) elif all(isinstance(x, np.ndarray) for x in [count, mean, m2]): with np.errstate(invalid="ignore"): - valid_counts: np.ndarray = count >= 2 # type: ignore - variance: np.ndarray = m2 # type: ignore - variance[valid_counts] /= count[valid_counts] - 1 # type: ignore - variance[np.invert(valid_counts)] = 0.0 + valid_counts: np.ndarray = count >= 2 # type: ignore + variance: np.ndarray = m2 # type: ignore + variance[valid_counts] /= count[valid_counts] - 1 # type: ignore + variance[np.invert(valid_counts)] = 0.0 return count, np.nan_to_num(mean), np.nan_to_num(np.sqrt(variance)) else: raise ValueError( diff --git a/mondey_backend/tests/utils/test_statistics.py b/mondey_backend/tests/utils/test_statistics.py index 8d0fe2ab..014b5699 100644 --- a/mondey_backend/tests/utils/test_statistics.py +++ b/mondey_backend/tests/utils/test_statistics.py @@ -2,6 +2,8 @@ import numpy as np import pytest +from sqlmodel import select + from mondey_backend.models.milestones import MilestoneAnswer from mondey_backend.models.milestones import MilestoneGroup from mondey_backend.routers.statistics import _add_sample @@ -9,7 +11,6 @@ from mondey_backend.routers.statistics import _get_statistics_by_age from mondey_backend.routers.statistics import calculate_milestone_statistics_by_age from mondey_backend.routers.statistics import calculate_milestonegroup_statistics_by_age -from sqlmodel import select def test_online_statistics_computation():