Remove 150K data rows limitation

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/dataset.py

@@ -606,11 +606,6 @@ def upsert_data_rows(
         >>>     ])
         >>>     task.wait_till_done()
         """
-        if len(items) > MAX_DATAROW_PER_API_OPERATION:
-            raise MalformedQueryException(
-                f"Cannot upsert more than {MAX_DATAROW_PER_API_OPERATION} DataRows per function call."
-            )
-
         specs = DataRowUpsertItem.build(self.uid, items)
         return self._exec_upsert_data_rows(specs, file_upload_thread_count)
 

libs/labelbox/tests/data/annotation_import/test_model_run.py

@@ -174,6 +174,7 @@ def test_model_run_split_assignment_by_data_row_ids(
     data_rows = dataset.create_data_rows([{
         "row_data": image_url
     } for _ in range(n_data_rows)])
+    data_rows.wait_till_done()
     data_row_ids = [data_row['id'] for data_row in data_rows.result]
     configured_project_with_one_data_row._wait_until_data_rows_are_processed(
         data_row_ids=data_row_ids)