fix tests

OpenMined · Oct 8, 2024 · 9ef5386 · 9ef5386
1 parent 8733333
commit 9ef5386
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/packages/syft/src/syft/service/sync/diff_state.py b/packages/syft/src/syft/service/sync/diff_state.py
@@ -1602,7 +1602,9 @@ def from_batch_decision(
                             )
                         ]
                     }
-                    if diff.obj_type in [Job, SyftLog, ActionObject, Request]:
+                    if diff.obj_type in [Job, SyftLog, Request] or issubclass(
+                        diff.obj_type, ActionObject
+                    ):
                         new_permissions_high_side = {
                             diff.obj_type: [
                                 ActionObjectPermission(

diff --git a/packages/syft/src/syft/service/sync/sync_service.py b/packages/syft/src/syft/service/sync/sync_service.py
@@ -210,7 +210,9 @@ def sync_items(
             for permission in permission_list:
                 if permission.uid in item_ids:
                     continue
-                if obj_type not in [Job, SyftLog, ActionObject, Request]:
+                if obj_type not in [Job, SyftLog, Request] and not issubclass(
+                    obj_type, ActionObject
+                ):
                     raise SyftException(
                         public_message="Permission for object type not supported!"
                     )