Add pull method to ExclusiveRemoteTask

src/bloqade/analog/task/exclusive.py

@@ -2,6 +2,7 @@
 import abc
 import uuid
 import re
+import time
 
 from beartype.typing import Dict
 from dataclasses import dataclass, field
@@ -245,12 +246,33 @@
 
         return self
 
-    def pull(self):
-        # Please avoid using this method, it's blocking and the waiting time is hours long
-        # Throw an error saying this is not supported
-        raise NotImplementedError(
-            "Pulling is not supported. Please use fetch() instead."
-        )
+    def pull(self, poll_interval: float = 20):
+        """
+        Blocking pull to get the task result.
+        poll_interval is the time interval to poll the task status.
+        Please ensure that it is relatively large, otherwise
+        the server could get overloaded with queries.
+        """
+
+        while True:
+            if self._task_result_ir.task_status is QuEraTaskStatusCode.Unsubmitted:
+                raise ValueError("Task ID not found.")
+
+            if self._task_result_ir.task_status in [
+                QuEraTaskStatusCode.Completed,
+                QuEraTaskStatusCode.Partial,
+                QuEraTaskStatusCode.Failed,
+                QuEraTaskStatusCode.Unaccepted,
+                QuEraTaskStatusCode.Cancelled,
+            ]:
+                return self
+
+            status = self.status()
+            if status in [QuEraTaskStatusCode.Completed, QuEraTaskStatusCode.Partial]:
+                self._task_result_ir = self._http_handler.fetch_results(self._task_id)
+                return self
+
+            time.sleep(poll_interval)
 
     def cancel(self):
         # This is not supported