[AL-5519] Add SDK Validation (#1053)

Author: mnoszczak

labelbox/schema/data_row.py

@@ -7,7 +7,7 @@
 from labelbox.orm.db_object import DbObject, Updateable, BulkDeletable
 from labelbox.orm.model import Entity, Field, Relationship
 from labelbox.schema.data_row_metadata import DataRowMetadataField  # type: ignore
-from labelbox.schema.export_params import CatalogExportParams
+from labelbox.schema.export_params import CatalogExportParams, validate_catalog_export_params
 from labelbox.schema.task import Task
 from labelbox.schema.user import User  # type: ignore
 
@@ -177,7 +177,6 @@ def export_v2(client: 'Client',
         >>>     task.wait_till_done()
         >>>     task.result
         """
-        print('export start')
 
         _params = params or CatalogExportParams({
             "attachments": False,
@@ -191,6 +190,8 @@ def export_v2(client: 'Client',
             "project_ids": None,
         })
 
+        validate_catalog_export_params(_params)
+
         mutation_name = "exportDataRowsInCatalog"
         create_task_query_str = """mutation exportDataRowsInCatalogPyApi($input: ExportDataRowsInCatalogInput!){
             %s(input: $input) {taskId} }

labelbox/schema/dataset.py

@@ -19,7 +19,7 @@
 from labelbox.exceptions import MalformedQueryException
 from labelbox.schema.data_row import DataRow
 from labelbox.schema.export_filters import DatasetExportFilters, SharedExportFilters
-from labelbox.schema.export_params import CatalogExportParams
+from labelbox.schema.export_params import CatalogExportParams, validate_catalog_export_params
 from labelbox.schema.project import _validate_datetime
 from labelbox.schema.task import Task
 from labelbox.schema.user import User
@@ -570,6 +570,7 @@ def export_v2(self,
             "model_run_ids": None,
             "project_ids": None,
         })
+        validate_catalog_export_params(_params)
 
         _filters = filters or DatasetExportFilters({
             "last_activity_at": None,

labelbox/schema/export_params.py

@@ -2,6 +2,8 @@
 
 from typing import Optional, List
 
+EXPORT_LIMIT = 30
+
 from labelbox.schema.media_type import MediaType
 if sys.version_info >= (3, 8):
     from typing import TypedDict
@@ -32,4 +34,19 @@ class CatalogExportParams(DataRowParams):
 
 
 class ModelRunExportParams(DataRowParams):
-    predictions: Optional[bool]
+    predictions: Optional[bool]
+
+
+def _validate_array_length(array, max_length, array_name):
+    if len(array) > max_length:
+        raise ValueError(f"{array_name} cannot exceed {max_length} items")
+
+
+def validate_catalog_export_params(params: CatalogExportParams):
+    if "model_run_ids" in params and params["model_run_ids"] is not None:
+        _validate_array_length(params["model_run_ids"], EXPORT_LIMIT,
+                               "model_run_ids")
+
+    if "project_ids" in params and params["project_ids"] is not None:
+        _validate_array_length(params["project_ids"], EXPORT_LIMIT,
+                               "project_ids")

labelbox/schema/slice.py

@@ -3,7 +3,7 @@
 from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Entity, Field
 from labelbox.pagination import PaginatedCollection
-from labelbox.schema.export_params import CatalogExportParams
+from labelbox.schema.export_params import CatalogExportParams, validate_catalog_export_params
 from labelbox.schema.task import Task
 from labelbox.schema.user import User
 
@@ -88,6 +88,7 @@ def export_v2(self,
             "model_run_ids": None,
             "project_ids": None,
         })
+        validate_catalog_export_params(_params)
 
         mutation_name = "exportDataRowsInSlice"
         create_task_query_str = """mutation exportDataRowsInSlicePyApi($input: ExportDataRowsInSliceInput!){