From 536c29978dbb5736b1281d512e8d727970abe68c Mon Sep 17 00:00:00 2001 From: Guilherme Schardong Date: Wed, 6 May 2026 16:28:28 +0100 Subject: [PATCH] `Capability` enum change COLOR=>TONE. --- src/facebias/estimators/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/facebias/estimators/__init__.py b/src/facebias/estimators/__init__.py index f369c98..c78136b 100644 --- a/src/facebias/estimators/__init__.py +++ b/src/facebias/estimators/__init__.py @@ -12,7 +12,7 @@ class Capability(StrEnum): AGE = "age" AGEGROUP = "age_group" SEX = "sex" - SKINCOLOR = "skin_color" + SKINTONE = "skin_tone" ETHNICITY = "ethinicity" @@ -23,7 +23,9 @@ class InvalidCapabilityError(Exception): class BaseEstimator(ABC): @abstractmethod - def predict(self, images: dict[Path, np.ndarray]) -> dict[Path, dict[Capability, Any]]: + def predict( + self, images: dict[Path, np.ndarray] + ) -> dict[Path, dict[Capability, Any]]: """Runs the estimator on a batch of images. The input `images` is a dictionary indexed by the image name, or ID,