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,