Fix BaseStash.update typing

OpenMined · Jan 16, 2024 · 2b2823c · 2b2823c
1 parent d4bc2a4
commit 2b2823c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/packages/syft/src/syft/service/worker/worker_service.py b/packages/syft/src/syft/service/worker/worker_service.py
@@ -3,7 +3,6 @@
 from typing import List
 from typing import Tuple
 from typing import Union
-from typing import cast
 
 # third party
 import docker
@@ -156,5 +155,5 @@ def _check_and_update_status_for_worker(
             message=f"Failed to update status for worker: {worker.id}. Error: {result.err()}"
         )
         if result.is_err()
-        else cast(SyftWorker, result.ok())
+        else result.ok()
     )
diff --git a/packages/syft/src/syft/store/document_store.py b/packages/syft/src/syft/store/document_store.py
@@ -699,7 +699,7 @@ def update(
         credentials: SyftVerifyKey,
         obj: BaseStash.object_type,
         has_permission=False,
-    ) -> Optional[Result[BaseStash.object_type, str]]:
+    ) -> Result[BaseStash.object_type, str]:
         qk = self.partition.store_query_key(obj)
         return self.partition.update(
             credentials=credentials, qk=qk, obj=obj, has_permission=has_permission