camille-vanhoffelen
fix: model_type filtering bug
4d00187
raw
history blame contribute delete
372 Bytes
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"