Fix unit/notebook tests

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

@@ -288,6 +288,7 @@ def reconstruct_args_kwargs(
             final_kwargs[param_key] = param.default
         else:
             raise Exception(f"Missing {param_key} not in kwargs.")
+    final_kwargs['context'] = kwargs['context'] if 'context' in kwargs else None
     return (args, final_kwargs)
 
 
@@ -340,7 +341,6 @@ def wrapper(func: Any) -> Callable:
         _path = class_name + "." + func_name
         signature = inspect.signature(func)
         signature = signature_remove_self(signature)
-        signature_with_context = deepcopy(signature)
         signature = signature_remove_context(signature)
 
         input_signature = deepcopy(signature)
@@ -354,7 +354,7 @@ def _decorator(self: Any, *args: Any, **kwargs: Any) -> Callable:
                 )
             if autosplat is not None and len(autosplat) > 0:
                 args, kwargs = reconstruct_args_kwargs(
-                    signature=signature_with_context,
+                    signature=input_signature,
                     autosplat=autosplat,
                     args=args,
                     kwargs=kwargs,