fix ActionObject.get_sync_dependencies

OpenMined · Oct 1, 2024 · 454ecbf · 454ecbf
1 parent e68ae99
commit 454ecbf
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/packages/syft/src/syft/service/action/action_object.py b/packages/syft/src/syft/service/action/action_object.py
@@ -1171,16 +1171,11 @@ def syft_make_action_with_self(
 
     def get_sync_dependencies(
         self, context: AuthedServiceContext, **kwargs: dict
-    ) -> list[UID]:  # type: ignore
-        # relative
-        from ..job.job_stash import Job
-
-        job: Job | None = context.server.services.job.get_by_result_id(
-            context, self.id.id
-        )  # type: ignore
-        if job is not None:
+    ) -> list[UID]:
+        try:
+            job = context.server.services.job.get_by_result_id(context, self.id.id)
             return [job.id]
-        else:
+        except SyftException:
             return []
 
     def syft_get_path(self) -> str: