diff --git a/playground/infrastructure/models.py b/playground/infrastructure/models.py index 0c7311e0f77b..a6beeb3e58fc 100644 --- a/playground/infrastructure/models.py +++ b/playground/infrastructure/models.py @@ -199,7 +199,7 @@ def dataset_defined(cls, v, values, **kwargs): @validator("datasets") def dataset_file_name(cls, datasets): for dataset_id, dataset in datasets.items(): - dataset.file_name = f"{dataset_id}.{dataset.format}" + dataset.file_name = f"{dataset_id}.{dataset.format.value}" if dataset.location == DatasetLocation.LOCAL: dataset_path = os.path.join( RepoProps.REPO_DATASETS_PATH, dataset.file_name