From e09ae8e252e6c1becad881c4696cfff36ca9cee8 Mon Sep 17 00:00:00 2001 From: John Mulligan Date: Tue, 18 Mar 2025 15:56:25 -0400 Subject: [PATCH] reef: mgr/diskprediction_local: avoid more mypy errors Similar to c4111033172db28c4737e8438f27901811919ce4 this patch suppresses mypy errors in the diskprediction_local mgr module. I probably put the magic comment on more lines than needed but mypy does not have a block-comment method to suppress checking for just a region of code today. This patch is not a backport as the issue is only impacting reef CI jobs and so it is applied directly to the reef branch. Signed-off-by: John Mulligan --- src/pybind/mgr/diskprediction_local/predictor.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/pybind/mgr/diskprediction_local/predictor.py b/src/pybind/mgr/diskprediction_local/predictor.py index 9ce57bf951e16..fa38165b0e5f2 100644 --- a/src/pybind/mgr/diskprediction_local/predictor.py +++ b/src/pybind/mgr/diskprediction_local/predictor.py @@ -168,14 +168,14 @@ class RHDiskFailurePredictor(Predictor): roll_window_size = 6 # rolling means generator - dataset_size = disk_days_attrs.shape[0] - roll_window_size + 1 - gen = (disk_days_attrs[i: i + roll_window_size, ...].mean(axis=0) - for i in range(dataset_size)) + dataset_size = disk_days_attrs.shape[0] - roll_window_size + 1 # type:ignore + gen = (disk_days_attrs[i: i + roll_window_size, ...].mean(axis=0) # type:ignore + for i in range(dataset_size)) # type:ignore means = np.vstack(gen) # type: ignore # rolling stds generator - gen = (disk_days_attrs[i: i + roll_window_size, ...].std(axis=0, ddof=1) - for i in range(dataset_size)) + gen = (disk_days_attrs[i: i + roll_window_size, ...].std(axis=0, ddof=1) # type: ignore + for i in range(dataset_size)) # type:ignore stds = np.vstack(gen) # type: ignore # coefficient of variation -- 2.39.5