From 331881ecf181662025fed22b5b8841cd184b644c Mon Sep 17 00:00:00 2001 From: teo Date: Mon, 14 Oct 2024 10:06:50 +0300 Subject: [PATCH] fix test --- packages/syft/src/syft/service/migration/migration_service.py | 4 ++-- packages/syft/src/syft/types/syft_object.py | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) 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__)