fix test

OpenMined · Oct 14, 2024 · 331881e · 331881e
1 parent 1fef3ed
commit 331881e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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__)