"""Constants shared across the benchmarks.""" class DatasetKey: TRAIN = "train" VALID = "val" TEST = "test"