Fix SDK build tests

Author: Kevin Kim

labelbox/client.py

@@ -1178,9 +1178,13 @@ def _format_failed_rows(rows: List[str],
                 errors.extend(
                     _format_failed_rows(data['deletedDataRowGlobalKeys'],
                                         "Data Row deleted"))
+
+                # Invalid results may contain empty string, so we must filter
+                # them prior to checking for PARTIAL_SUCCESS
+                filtered_results = list(filter(lambda r: r != '', results))
                 if not errors:
                     status = CollectionJobStatus.SUCCESS.value
-                elif errors and results:
+                elif errors and len(filtered_results) > 0:
                     status = CollectionJobStatus.PARTIAL_SUCCESS.value
                 else:
                     status = CollectionJobStatus.FAILURE.value

labelbox/schema/task.py

@@ -63,9 +63,7 @@ def wait_till_done(self, timeout_seconds=300) -> None:
         check_frequency = 2  # frequency of checking, in seconds
         while True:
             if self.status != "IN_PROGRESS":
-                if self.status == "FAILED" or (self.status == "COMPLETE" and
-                                               self.failed_data_rows
-                                               is not None):
+                if self.errors is not None:
                     logger.warning(
                         "There are errors present. Please look at `task.errors` for more details"
                     )
@@ -87,7 +85,7 @@ def errors(self) -> Optional[Dict[str, Any]]:
             result = self._fetch_remote_json()
             return result["error"]
         elif self.status == "COMPLETE":
-            return self.failed_data_rows()
+            return self.failed_data_rows
         return None
 
     @property
@@ -105,11 +103,12 @@ def result(self) -> List[Dict[str, Any]]:
                 'global_key': data_row.get('globalKey'),
             } for data_row in result['createdDataRows']]
 
+    @property
     def failed_data_rows(self) -> Optional[Dict[str, Any]]:
         """ Fetch data rows which failed to be created for an import task.
         """
         result = self._fetch_remote_json()
-        if result.get("errors") is not None:
+        if len(result.get("errors", [])) > 0:
             return result["errors"]
         else:
             return None

tests/integration/test_data_rows.py

@@ -352,7 +352,8 @@ def test_create_data_rows_with_invalid_metadata(dataset, image_url):
         DataRow.metadata_fields: fields
     }])
     task.wait_till_done()
-    assert task.status == "FAILED"
+    assert task.status == "COMPLETE"
+    assert len(task.failed_data_rows) > 0
 
 
 def test_create_data_rows_with_metadata_missing_value(dataset, image_url):
@@ -632,7 +633,8 @@ def test_data_row_bulk_creation_with_same_global_keys(dataset, sample_image):
     }])
 
     task.wait_till_done()
-    assert task.status == "FAILED"
+    assert task.status == "COMPLETE"
+    assert len(task.failed_data_rows) > 0
     assert len(list(dataset.data_rows())) == 0
 
     task = dataset.create_data_rows([{

tests/integration/test_global_keys.py

@@ -251,9 +251,10 @@ def test_get_data_row_ids_for_invalid_global_keys(client, dataset, image_url):
     assert res['status'] == "PARTIAL SUCCESS"
 
     assert len(res['errors']) == 1
-    assert len(res['results']) == 1
+    assert len(res['results']) == 2
 
     assert res['errors'][0]['error'] == "Data Row not found"
     assert res['errors'][0]['global_key'] == gk_1
 
-    assert res['results'][0] == dr_2.uid
+    assert res['results'][0] == ''
+    assert res['results'][1] == dr_2.uid