[bugfix] fix KeyError: 'context' in reconstruct_args_kwargs

OpenMined · May 28, 2024 · cd44055 · cd44055
1 parent 009123e
commit cd44055
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/syft/src/syft/service/service.py b/packages/syft/src/syft/service/service.py
@@ -289,7 +289,7 @@ def reconstruct_args_kwargs(
         else:
             raise Exception(f"Missing {param_key} not in kwargs.")
 
-    if "context":
+    if "context" in kwargs:
         final_kwargs["context"] = kwargs["context"]
 
     return (args, final_kwargs)