[syft/action_service] only add blob permission if the entry's id is n…

…ot None
OpenMined · Jul 5, 2024 · 787af9b · 787af9b
1 parent 54ba6b3
commit 787af9b
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/packages/syft/src/syft/service/action/action_service.py b/packages/syft/src/syft/service/action/action_service.py
@@ -565,14 +565,21 @@ def store_permission(
 
  def blob_permission(
  x: SyftVerifyKey | None = None,
- ) -> ActionObjectPermission:
- return ActionObjectPermission(result_blob_id, read_permission, x)
+ ) -> ActionObjectPermission | None:
+ if result_blob_id:
+ return ActionObjectPermission(result_blob_id, read_permission, x)
+ else:
+ return None
 
  if len(output_readers) > 0:
  store_permissions = [store_permission(x) for x in output_readers]
  self.store.add_permissions(store_permissions)
 
- blob_permissions = [blob_permission(x) for x in output_readers]
+ blob_permissions = [
+ blob_permission(x)
+ for x in output_readers
+ if blob_permission(x) is not None
+ ]
  blob_storage_service.stash.add_permissions(blob_permissions)
 
  return set_result