Capability enum change COLOR=>TONE.

This commit is contained in:
2026-05-06 16:28:28 +01:00
parent e2790d2d5d
commit 536c29978d

View File

@@ -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,