update status

OpenMined · May 21, 2024 · 49507ca · 49507ca
1 parent 32bd5d5
commit 49507ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/syft/src/syft/service/job/job_stash.py b/packages/syft/src/syft/service/job/job_stash.py
@@ -644,6 +644,10 @@ def _repr_html_(self) -> str:
     def wait(
         self, job_only: bool = False, timeout: int | None = None
     ) -> Any | SyftNotReady:
+        self.fetch()
+        if self.resolved:
+            return self.resolve
+
         api = APIRegistry.api_for(
             node_uid=self.syft_node_location,
             user_verify_key=self.syft_client_verify_key,
@@ -652,8 +656,6 @@ def wait(
             raise ValueError(
                 f"Can't access Syft API. You must login to {self.syft_node_location}"
             )
-        if self.resolved:
-            return self.resolve
 
         workers = api.services.worker.get_all()
         if not workers: