From cd440556eacd4c3c592bf4ea538884df9b638715 Mon Sep 17 00:00:00 2001
From: khoaguin <dkn.work@protonmail.com>
Date: Tue, 28 May 2024 09:57:44 +0700
Subject: [PATCH] [bugfix] fix `KeyError: 'context'` in
 `reconstruct_args_kwargs`

---
 packages/syft/src/syft/service/service.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/syft/src/syft/service/service.py b/packages/syft/src/syft/service/service.py
index 55f2c1f4b5d..cda115cb8b4 100644
--- 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)