Fix an issue when unpickling functions through the escape hatch (#1770)

Netflix · Mar 23, 2024 · 57640fd · 57640fd
1 parent fbe4aa1
commit 57640fd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/metaflow/plugins/env_escape/client.py b/metaflow/plugins/env_escape/client.py
@@ -386,7 +386,13 @@ def name_to_parent_name(name):
                 raise RuntimeError("Local function unpickling without an object ID")
             if obj_id not in self._proxied_standalone_functions:
                 self._proxied_standalone_functions[obj_id] = create_class(
-                    self, "__function_%s" % obj_id, {}, {}, {}, {"__call__": ""}, []
+                    self,
+                    "__main__.__function_%s" % obj_id,
+                    {},
+                    {},
+                    {},
+                    {"__call__": ""},
+                    [],
                 )
             return self._proxied_standalone_functions[obj_id]
         local_class = self._proxied_classes.get(lookup_name, None)