[X-0] Add field definitions (#1745)

Author: kozikkamil

libs/labelbox/src/labelbox/schema/project.py

@@ -83,7 +83,7 @@ def validate_labeling_parameter_overrides(
 
 
 class Project(DbObject, Updateable, Deletable):
-    """ A Project is a container that includes a labeling frontend, an ontology,
+    """A Project is a container that includes a labeling frontend, an ontology,
     datasets and labels.
 
     Attributes:
@@ -96,6 +96,8 @@ class Project(DbObject, Updateable, Deletable):
         queue_mode (string)
         auto_audit_number_of_labels (int)
         auto_audit_percentage (float)
+        is_benchmark_enabled (bool)
+        is_consensus_enabled (bool)
 
         created_by (Relationship): `ToOne` relationship to User
         organization (Relationship): `ToOne` relationship to Organization
@@ -123,6 +125,8 @@ class Project(DbObject, Updateable, Deletable):
     data_row_count = Field.Int("data_row_count")
     model_setup_complete: Field = Field.Boolean("model_setup_complete")
     upload_type: Field = Field.Enum(UploadType, "upload_type")
+    is_benchmark_enabled = Field.Boolean("is_benchmark_enabled")
+    is_consensus_enabled = Field.Boolean("is_consensus_enabled")
 
     # Relationships
     created_by = Relationship.ToOne("User", False, "created_by")

libs/labelbox/tests/integration/test_legacy_project.py

@@ -17,16 +17,14 @@ def test_project_dataset(client, rand_gen):
 
 def test_project_auto_audit_parameters(client, rand_gen):
     with pytest.raises(
-            ValueError,
-            match=
-            "quality_mode must be set instead of auto_audit_percentage or auto_audit_number_of_labels."
+        ValueError,
+        match="quality_modes must be set instead of auto_audit_percentage or auto_audit_number_of_labels.",
     ):
         client.create_project(name=rand_gen(str), auto_audit_percentage=0.5)
 
     with pytest.raises(
-            ValueError,
-            match=
-            "quality_mode must be set instead of auto_audit_percentage or auto_audit_number_of_labels."
+        ValueError,
+        match="quality_modes must be set instead of auto_audit_percentage or auto_audit_number_of_labels.",
     ):
         client.create_project(name=rand_gen(str), auto_audit_number_of_labels=2)
 

libs/labelbox/tests/unit/test_project.py

@@ -8,7 +8,8 @@
 @pytest.fixture
 def project_entity():
     return Project(
-        MagicMock(), {
+        MagicMock(),
+        {
             "id": "test",
             "name": "test",
             "createdAt": "2021-06-01T00:00:00.000Z",
@@ -24,7 +25,10 @@ def project_entity():
             "setupComplete": "2021-06-01T00:00:00.000Z",
             "modelSetupComplete": None,
             "uploadType": "Auto",
-        })
+            "isBenchmarkEnabled": False,
+            "isConsensusEnabled": False,
+        },
+    )
 
 
 @pytest.mark.parametrize(
@@ -39,7 +43,8 @@ def test_project_editor_task_type(api_editor_task_type,
                                   expected_editor_task_type, project_entity):
     client = MagicMock()
     project = Project(
-        client, {
+        client,
+        {
             "id": "test",
             "name": "test",
             "createdAt": "2021-06-01T00:00:00.000Z",
@@ -55,7 +60,10 @@ def test_project_editor_task_type(api_editor_task_type,
             "setupComplete": "2021-06-01T00:00:00.000Z",
             "modelSetupComplete": None,
             "uploadType": "Auto",
-        })
+            "isBenchmarkEnabled": False,
+            "isConsensusEnabled": False,
+        },
+    )
 
     assert project.editor_task_type == expected_editor_task_type