safer pop from working namespace

log failure as a warning because it's weird, but not impossible
ipython · Oct 30, 2024 · 59b5a3f · 59b5a3f
1 parent 7742389
commit 59b5a3f
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/ipyparallel/engine/kernel.py b/ipyparallel/engine/kernel.py
@@ -139,12 +139,10 @@ def do_apply(self, content, bufs, msg_id, reply_metadata):
         try:
             working = shell.user_ns
 
-            prefix = "_" + str(msg_id).replace("-", "") + "_"
+            prefix = f"_{str(msg_id).replace('-', '_')}_"
 
             f, args, kwargs = unpack_apply_message(bufs, working, copy=False)
 
-            fname = getattr(f, '__name__', 'f')
-
             fname = prefix + "f"
             argname = prefix + "args"
             kwargname = prefix + "kwargs"
@@ -153,13 +151,18 @@ def do_apply(self, content, bufs, msg_id, reply_metadata):
             ns = {fname: f, argname: args, kwargname: kwargs, resultname: None}
             # print ns
             working.update(ns)
-            code = f"{resultname} = {fname}(*{argname},**{kwargname})"
+            code = f"{resultname} = {fname}(*{argname}, **{kwargname})"
             try:
                 exec(code, shell.user_global_ns, shell.user_ns)
                 result = working.get(resultname)
             finally:
                 for key in ns:
-                    working.pop(key)
+                    try:
+                        working.pop(key)
+                    except KeyError:
+                        self.log.warning(
+                            f"Failed to undefine temporary apply variable {key}, already undefined"
+                        )
 
             result_buf = serialize_object(
                 result,