from enum import StrEnum


class BenchmarkDataset(StrEnum):
    """Identifiers for each dataset in the benchmark suite.

    Values correspond to the JSON result filenames (without extension).
    """

    OPENPII = "ai4privacy__pii-masking-openpii-1m"
    GRETEL = "gretelai__gretel-pii-masking-en-v1"
    NEMOTRON_PII = "nvidia__Nemotron-PII"
    PRIVY = "piimb__privy"