File size: 372 Bytes
aa36ef8 4d00187 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | 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"
|