yapf

Author: jtsodapop

labelbox/schema/batch.py

@@ -77,7 +77,9 @@ def remove_queued_data_rows(self) -> None:
                 },
                             experimental=True)
 
-    def export_data_rows(self, timeout_seconds=120, include_metadata: bool=False) -> Generator:
+    def export_data_rows(self,
+                         timeout_seconds=120,
+                         include_metadata: bool = False) -> Generator:
         """ Returns a generator that produces all data rows that are currently
         in this batch.
 
@@ -98,18 +100,21 @@ def export_data_rows(self, timeout_seconds=120, include_metadata: bool=False) ->
         """ % (id_param, metadata_param, id_param, metadata_param)
         sleep_time = 2
         while True:
-            res = self.client.execute(query_str, {id_param: self.uid, metadata_param: include_metadata})
+            res = self.client.execute(query_str, {
+                id_param: self.uid,
+                metadata_param: include_metadata
+            })
             res = res["exportBatchDataRows"]
             if res["status"] == "COMPLETE":
                 download_url = res["downloadUrl"]
                 response = requests.get(download_url)
                 response.raise_for_status()
                 reader = ndjson.reader(StringIO(response.text))
-                return (Entity.DataRow(self.client, {
-                    **result, 
-                    'customMetadata': result['metadata'],
-                    'metadataFields': result['metadataFields']
-                }) for result in reader)
+                return (Entity.DataRow(
+                    self.client, {
+                        **result, 'customMetadata': result['metadata'],
+                        'metadataFields': result['metadataFields']
+                    }) for result in reader)
             elif res["status"] == "FAILED":
                 raise LabelboxError("Data row export failed.")
 

labelbox/schema/dataset.py

@@ -462,7 +462,9 @@ def data_row_for_external_id(self, external_id) -> "DataRow":
                 external_id)
         return data_rows[0]
 
-    def export_data_rows(self, timeout_seconds=120, include_metadata: bool=False) -> Generator:
+    def export_data_rows(self,
+                         timeout_seconds=120,
+                         include_metadata: bool = False) -> Generator:
         """ Returns a generator that produces all data rows that are currently
         attached to this dataset.
 
@@ -483,18 +485,21 @@ def export_data_rows(self, timeout_seconds=120, include_metadata: bool=False) ->
         """ % (id_param, metadata_param, id_param, metadata_param)
         sleep_time = 2
         while True:
-            res = self.client.execute(query_str, {id_param: self.uid, metadata_param: include_metadata})
+            res = self.client.execute(query_str, {
+                id_param: self.uid,
+                metadata_param: include_metadata
+            })
             res = res["exportDatasetDataRows"]
             if res["status"] == "COMPLETE":
                 download_url = res["downloadUrl"]
                 response = requests.get(download_url)
                 response.raise_for_status()
                 reader = ndjson.reader(StringIO(response.text))
-                return (Entity.DataRow(self.client, {
-                    **result, 
-                    'customMetadata': result['metadata'],
-                    'metadataFields': result['metadataFields']
-                }) for result in reader)
+                return (Entity.DataRow(
+                    self.client, {
+                        **result, 'customMetadata': result['metadata'],
+                        'metadataFields': result['metadataFields']
+                    }) for result in reader)
             elif res["status"] == "FAILED":
                 raise LabelboxError("Data row export failed.")
 
@@ -507,5 +512,3 @@ def export_data_rows(self, timeout_seconds=120, include_metadata: bool=False) ->
             logger.debug("Dataset '%s' data row export, waiting for server...",
                          self.uid)
             time.sleep(sleep_time)
-
-

labelbox/schema/project.py

@@ -185,8 +185,10 @@ def labels(self, datasets=None, order_by=None) -> PaginatedCollection:
         return PaginatedCollection(self.client, query_str, {id_param: self.uid},
                                    ["project", "labels"], Label)
 
-    def export_queued_data_rows(self,
-                                timeout_seconds=120,  include_metadata: bool=False) -> List[Dict[str, str]]:
+    def export_queued_data_rows(
+            self,
+            timeout_seconds=120,
+            include_metadata: bool = False) -> List[Dict[str, str]]:
         """ Returns all data rows that are currently enqueued for this project.
 
         Args:
@@ -203,7 +205,10 @@ def export_queued_data_rows(self,
         """ % (id_param, metadata_param, id_param, metadata_param)
         sleep_time = 2
         while True:
-            res = self.client.execute(query_str, {id_param: self.uid, metadata_param: include_metadata})
+            res = self.client.execute(query_str, {
+                id_param: self.uid,
+                metadata_param: include_metadata
+            })
             res = res["exportQueuedDataRows"]
             if res["status"] == "COMPLETE":
                 download_url = res["downloadUrl"]