Merge pull request #8767 from OpenMined/eelco/workerpool-error-handling

add error handling for get workerpool
OpenMined · Apr 30, 2024 · f060e77 · f060e77
2 parents a6ce0a7 + 97bd314
commit f060e77
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/packages/syft/src/syft/node/node.py b/packages/syft/src/syft/node/node.py
@@ -1230,6 +1230,11 @@ def add_api_endpoint_execution_to_queue(
         else:
             worker_pool = self.get_worker_pool_by_name(worker_pool)
 
+        if isinstance(worker_pool, SyftError):
+            return worker_pool
+        elif worker_pool is None:
+            return SyftError(message="Worker pool not found")
+
         # Create a Worker pool reference object
         worker_pool_ref = LinkedObject.from_obj(
             worker_pool,