Allow to pass batch consensus settings when creating a batch

Author: mkozik-lb

labelbox/schema/consensus_settings.py

@@ -0,0 +1,18 @@
+from pydantic import BaseModel
+
+
+class ConsensusSettings(BaseModel):
+    """Container for holding consensus quality settings
+
+    >>> ConsensusSettings(
+    >>>    numberOfLabels = 2,
+    >>>    coveragePercentage = 0.2
+    >>>  )
+
+    Args:
+        numberOfLabels: Number of labels for consensus
+        coveragePercentage: Percentage of data rows to be labeled more than once
+    """
+
+    numberOfLabels: int
+    coveragePercentage: float

labelbox/schema/project.py

@@ -16,6 +16,7 @@
 from labelbox.orm.db_object import DbObject, Updateable, Deletable
 from labelbox.orm.model import Entity, Field, Relationship
 from labelbox.pagination import PaginatedCollection
+from labelbox.schema.consensus_settings import ConsensusSettings
 from labelbox.schema.media_type import MediaType
 from labelbox.schema.queue_mode import QueueMode
 from labelbox.schema.resource_tag import ResourceTag
@@ -561,7 +562,11 @@ def setup(self, labeling_frontend, labeling_frontend_options) -> None:
         timestamp = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
         self.update(setup_complete=timestamp)
 
-    def create_batch(self, name: str, data_rows: List[str], priority: int = 5):
+    def create_batch(self,
+                     name: str,
+                     data_rows: List[str],
+                     priority: int = 5,
+                     consensus_settings: Optional[ConsensusSettings] = None):
         """Create a new batch for a project. Batches is in Beta and subject to change
 
         Args:
@@ -603,9 +608,14 @@ def create_batch(self, name: str, data_rows: List[str], priority: int = 5):
         params = {
             "projectId": self.uid,
             "batchInput": {
-                "name": name,
-                "dataRowIds": dr_ids,
-                "priority": priority
+                "name":
+                    name,
+                "dataRowIds":
+                    dr_ids,
+                "priority":
+                    priority,
+                "consensusSettings":
+                    consensus_settings.dict() if consensus_settings else None
             }
         }