Chaging the tests once more since we were swallowing the original error message

Author: tytalus

tests/integration/test_data_rows.py

@@ -636,6 +636,13 @@ def test_data_row_bulk_creation_with_same_global_keys(dataset, sample_image):
     assert task.status == "FAILED"
     assert len(task.failed_data_rows) > 0
     assert len(list(dataset.data_rows())) == 0
+    errorMsg = (
+        "Failed to create data rows after 5 retries. "
+        "This chunk's data rows will not be imported. "
+        "Retry by appending them to the dataset. "
+        "Error: Data rows contain empty string or duplicate global keys, which are not allowed"
+     )
+    assert task.errors == errorMsg
 
     task = dataset.create_data_rows([{
         DataRow.row_data: sample_image,

tests/integration/test_task.py

@@ -1,6 +1,6 @@
 import pytest
 
-from labelbox import DataRow, Task
+from labelbox import DataRow
 from labelbox.schema.data_row_metadata import DataRowMetadataField
 
 EMBEDDING_SCHEMA_ID = "ckpyije740000yxdk81pbgjdc"
@@ -27,7 +27,6 @@ def test_task_errors(dataset, image_url):
     assert task.status == "FAILED"
     assert len(task.failed_data_rows) > 0
     assert task.errors is not None
-    assert task.errors == "Failed to create data rows after 5 retries. This chunk's data rows will not be imported. Retry by appending them to the dataset."
 
 def test_task_success_json(dataset, image_url):
     client = dataset.client