fix warnings in dataset transform + privateattr check

packages/syft/src/syft/service/code/user_code.py

@@ -118,7 +118,7 @@ class UserCodeStatusCollection(SyncableSyftObject):
     status_dict: dict[NodeIdentity, tuple[UserCodeStatus, str]] = {}
     user_code_link: LinkedObject
 
-    def get_diffs(self, ext_obj: Any) -> list[AttrDiff]:
+    def syft_get_diffs(self, ext_obj: Any) -> list[AttrDiff]:
         # relative
         from ...service.sync.diff_state import AttrDiff
 

packages/syft/src/syft/service/dataset/dataset.py

@@ -756,17 +756,17 @@ def create_and_store_twin(context: TransformContext) -> TransformContext:
 
 
 def infer_shape(context: TransformContext) -> TransformContext:
-    if context.output is not None and context.output["shape"] is None:
+    if context.output is None:
+        raise ValueError(f"{context}'s output is None. No transformation happened")
+    if context.output["shape"] is None:
         if context.obj is not None and not _is_action_data_empty(context.obj.mock):
             context.output["shape"] = get_shape_or_len(context.obj.mock)
-    else:
-        print(f"{context}'s output is None. No transformation happened")
     return context
 
 
 def set_data_subjects(context: TransformContext) -> TransformContext | SyftError:
     if context.output is None:
-        return SyftError(f"{context}'s output is None. No transformation happened")
+        raise ValueError(f"{context}'s output is None. No transformation happened")
     if context.node is None:
         return SyftError(
             "f{context}'s node is None, please log in. No trasformation happened"
@@ -796,7 +796,7 @@ def add_default_node_uid(context: TransformContext) -> TransformContext:
         if context.output["node_uid"] is None and context.node is not None:
             context.output["node_uid"] = context.node.id
     else:
-        print(f"{context}'s output is None. No transformation happened.")
+        raise ValueError(f"{context}'s output is None. No transformation happened")
     return context
 
 

packages/syft/src/syft/service/notification/notifications.py

@@ -151,7 +151,7 @@ def add_msg_creation_time(context: TransformContext) -> TransformContext:
     if context.output is not None:
         context.output["created_at"] = DateTime.now()
     else:
-        print(f"{context}'s output is None. No transformation happened.")
+        raise ValueError(f"{context}'s output is None. No transformation happened")
     return context
 
 

packages/syft/src/syft/service/policy/policy.py

@@ -572,7 +572,7 @@ def generate_unique_class_name(context: TransformContext) -> TransformContext:
         unique_name = f"{service_class_name}_{context.credentials}_{code_hash}"
         context.output["unique_name"] = unique_name
     else:
-        print(f"{context}'s output is None. No transformation happened.")
+        raise ValueError(f"{context}'s output is None. No transformation happened")
 
     return context
 
@@ -696,7 +696,7 @@ def compile_code(context: TransformContext) -> TransformContext:
                 + context.output["parsed_code"]
             )
     else:
-        print(f"{context}'s output is None. No transformation happened.")
+        raise ValueError(f"{context}'s output is None. No transformation happened")
 
     return context
 

packages/syft/src/syft/types/syft_object.py

@@ -579,9 +579,10 @@ def __post_init__(self) -> None:
     def _syft_set_validate_private_attrs_(self, **kwargs: Any) -> None:
         # Validate and set private attributes
         # https://github.com/pydantic/pydantic/issues/2105
+        annotations = typing.get_type_hints(self.__class__)
         for attr, decl in self.__private_attributes__.items():
             value = kwargs.get(attr, decl.get_default())
-            var_annotation = self.__annotations__.get(attr)
+            var_annotation = annotations.get(attr)
             if value is not PydanticUndefined:
                 if var_annotation is not None:
                     # Otherwise validate value against the variable annotation