diff --git a/packages/syft/src/syft/service/migration/migration_service.py b/packages/syft/src/syft/service/migration/migration_service.py index d4f217b9035..4a346ebde9d 100644 --- a/packages/syft/src/syft/service/migration/migration_service.py +++ b/packages/syft/src/syft/service/migration/migration_service.py @@ -500,7 +500,7 @@ def reset_and_restore( return SyftSuccess(message="Database reset successfully.") @service_method( - path="sync._get_object", + path="migration._get_object", name="_get_object", roles=DATA_SCIENTIST_ROLE_LEVEL, ) @@ -514,7 +514,7 @@ def _get_object( ) @service_method( - path="sync._update_object", + path="migration._update_object", name="_update_object", roles=ADMIN_ROLE_LEVEL, ) diff --git a/packages/syft/src/syft/types/syft_object.py b/packages/syft/src/syft/types/syft_object.py index a1a1c3236b7..1e93d377460 100644 --- a/packages/syft/src/syft/types/syft_object.py +++ b/packages/syft/src/syft/types/syft_object.py @@ -434,7 +434,9 @@ def refresh(self) -> None: api = self._get_api() except Exception as _: return - new_object = api.services.sync._get_object(uid=self.id, object_type=type(self)) + new_object = api.services.migration._get_object( + uid=self.id, object_type=type(self) + ) if type(new_object) == type(self): self.__dict__.update(new_object.__dict__)