diff --git a/chart/docker-images.yaml b/chart/docker-images.yaml index f1b0117902..047546f5c8 100644 --- a/chart/docker-images.yaml +++ b/chart/docker-images.yaml @@ -2,7 +2,7 @@ "dockerImage": { "reverseProxy": "docker.io/nginx:1.20", "jobs": { - "mongodbMigration": "huggingface/datasets-server-jobs-mongodb_migration:sha-3de1315" + "mongodbMigration": "huggingface/datasets-server-jobs-mongodb_migration:sha-48dee06" }, "services": { "admin": "huggingface/datasets-server-services-admin:sha-3de1315", diff --git a/jobs/mongodb_migration/src/mongodb_migration/migrations/_20230126164900_queue_job_add_priority.py b/jobs/mongodb_migration/src/mongodb_migration/migrations/_20230126164900_queue_job_add_priority.py index b73f135e04..43291b6f91 100644 --- a/jobs/mongodb_migration/src/mongodb_migration/migrations/_20230126164900_queue_job_add_priority.py +++ b/jobs/mongodb_migration/src/mongodb_migration/migrations/_20230126164900_queue_job_add_priority.py @@ -32,7 +32,7 @@ def validate(self) -> None: logging.info("Ensure that a random selection of jobs have the 'priority' field set to 'normal'") def custom_validation(doc: JobSnapshot) -> None: - if doc.priority != "normal": + if doc.priority != Priority.NORMAL: raise ValueError("priority should be 'normal'") check_documents(DocCls=JobSnapshot, sample_size=10, custom_validation=custom_validation)