Merge pull request #651 from corynezinstitchfix/label-export-timeout-fix

Fix timeout logic

Author: msokoloff1

labelbox/schema/project.py

@@ -205,6 +205,7 @@ def export_queued_data_rows(
             {exportQueuedDataRows(data:{projectId: $%s , includeMetadataInput: $%s}) {downloadUrl createdAt status} }
         """ % (id_param, metadata_param, id_param, metadata_param)
         sleep_time = 2
+        start_time = time.time()
         while True:
             res = self.client.execute(query_str, {
                 id_param: self.uid,
@@ -219,8 +220,8 @@ def export_queued_data_rows(
             elif res["status"] == "FAILED":
                 raise LabelboxError("Data row export failed.")
 
-            timeout_seconds -= sleep_time
-            if timeout_seconds <= 0:
+            current_time = time.time()
+            if current_time - start_time > timeout_seconds:
                 raise LabelboxError(
                     f"Unable to export data rows within {timeout_seconds} seconds."
                 )
@@ -328,6 +329,7 @@ def _validate_datetime(string_date: str) -> bool:
             {exportLabels(data:{projectId: $%s%s}) {downloadUrl createdAt shouldPoll} }
         """ % (id_param, id_param, filter_param)
 
+        start_time = time.time()
         while True:
             res = self.client.execute(query_str, {id_param: self.uid})
             res = res["exportLabels"]
@@ -340,8 +342,8 @@ def _validate_datetime(string_date: str) -> bool:
                     response.raise_for_status()
                     return response.json()
 
-            timeout_seconds -= sleep_time
-            if timeout_seconds <= 0:
+            current_time = time.time()
+            if current_time - start_time > timeout_seconds:
                 return None
 
             logger.debug("Project '%s' label export, waiting for server...",