From 5f25297aa86eb4deeef6a3054b59f13c990cd8b0 Mon Sep 17 00:00:00 2001 From: Richard Sun Date: Tue, 12 Sep 2023 15:25:13 -0700 Subject: [PATCH] [QQC-2474] Disallow invalid quality modes during create_project --- labelbox/client.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/labelbox/client.py b/labelbox/client.py index e19130baa..f6578cb9c 100644 --- a/labelbox/client.py +++ b/labelbox/client.py @@ -696,10 +696,12 @@ def create_project(self, **kwargs) -> Project: data[ "auto_audit_number_of_labels"] = BENCHMARK_AUTO_AUDIT_NUMBER_OF_LABELS data["auto_audit_percentage"] = BENCHMARK_AUTO_AUDIT_PERCENTAGE - else: + elif quality_mode is QualityMode.Consensus: data[ "auto_audit_number_of_labels"] = CONSENSUS_AUTO_AUDIT_NUMBER_OF_LABELS data["auto_audit_percentage"] = CONSENSUS_AUTO_AUDIT_PERCENTAGE + else: + raise ValueError(f"{quality_mode} is not a valid quality mode.") return self._create(Entity.Project, { **data,