From 14ab9abcc1fb4e1b7d71c490bd603660c4ab2ff2 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 24 Apr 2026 09:02:36 -0600 Subject: [PATCH] Fix mypy --- frigate/data_processing/common/face/model.py | 4 ++-- testing-scripts/{face_investigate.py => face_dataset.py} | 0 2 files changed, 2 insertions(+), 2 deletions(-) rename testing-scripts/{face_investigate.py => face_dataset.py} (100%) diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py index 8d8e169a65..87293f7f02 100644 --- a/frigate/data_processing/common/face/model.py +++ b/frigate/data_processing/common/face/model.py @@ -157,7 +157,7 @@ def build_class_mean( arr = np.stack(embs, axis=0) if len(arr) < 5: - return stats.trim_mean(arr, trim, axis=0) + return np.asarray(stats.trim_mean(arr, trim, axis=0)) keep = np.ones(len(arr), dtype=bool) floor = max(5, int(np.ceil(min_keep_frac * len(arr)))) @@ -185,7 +185,7 @@ def build_class_mean( f"Vector-wise outlier filter dropped {dropped}/{len(arr)} embeddings" ) - return stats.trim_mean(arr[keep], trim, axis=0) + return np.asarray(stats.trim_mean(arr[keep], trim, axis=0)) def similarity_to_confidence( diff --git a/testing-scripts/face_investigate.py b/testing-scripts/face_dataset.py similarity index 100% rename from testing-scripts/face_investigate.py rename to testing-scripts/face_dataset.py