Add result field to Task and get_result()

Author: Kevin Kim

labelbox/orm/db_object.py

@@ -150,6 +150,7 @@ def _to_many(self, where=None, order_by=None):
         query_string, params = query.relationship(
             self.source if self.filter_on_id else type(self.source), rel, where,
             order_by)
+        print(f"!! QUery string {query_string}")
         return PaginatedCollection(
             self.source.client, query_string, params,
             [utils.camel_case(self.source.type_name()), rel.graphql_name],

labelbox/schema/dataset.py

@@ -430,10 +430,11 @@ def export_data_rows(self, timeout_seconds=120) -> Generator:
                 response = requests.get(download_url)
                 response.raise_for_status()
                 reader = ndjson.reader(StringIO(response.text))
+                for r in reader:
+                    print(f"Reader: {r}")
                 # TODO: Update result to parse metadataFields when resolver returns
-                return (Entity.DataRow(self.client, {
-                    **result, 'metadataFields': []
-                }) for result in reader)
+                return (
+                    Entity.DataRow(self.client, result) for result in reader)
             elif res["status"] == "FAILED":
                 raise LabelboxError("Data row export failed.")
 

labelbox/schema/task.py

@@ -1,5 +1,8 @@
 import logging
+import requests
 import time
+import validators
+from validators import ValidationFailure
 from typing import TYPE_CHECKING, Optional
 
 from labelbox.exceptions import ResourceNotFoundError
@@ -31,6 +34,7 @@ class Task(DbObject):
     name = Field.String("name")
     status = Field.String("status")
     completion_percentage = Field.Float("completion_percentage")
+    result = Field.String("result")
     _user: Optional["User"] = None
 
     # Relationships
@@ -65,3 +69,15 @@ def wait_till_done(self, timeout_seconds=300) -> None:
             timeout_seconds -= check_frequency
             time.sleep(sleep_time_seconds)
             self.refresh()
+    
+    def get_result(self) -> str:
+        """ Downloads the result file from Task
+        """
+        if self.result:
+            result_url = validators.url(self.result)
+            if isinstance(result_url, ValidationFailure):
+                raise TypeError(f"{result_url} is not a valid url")
+            response = requests.get(self.result)
+            response.raise_for_status()
+            return response.text
+        return ""

requirements.txt

@@ -13,4 +13,5 @@ typeguard
 imagesize
 pyproj
 pygeotile
-typing-extensions
+typing-extensions
+validators