fix bug in mapping of function names in executor

Author: tphung3

parsl/executors/taskvine/executor.py

@@ -351,8 +351,7 @@ def submit(self, func, resource_specification, *args, **kwargs):
             exec_mode = 'regular'
 
         if exec_mode == 'serverless':
-            if func.__name__ not in self._map_func_names_to_func_details or\
-                    'func_obj' not in self._map_func_names_to_func_details[func.__name__]:
+            if func.__name__ not in self._map_func_names_to_func_details:
                 self._map_func_names_to_func_details[func.__name__] = {'func_obj': func}
             else:
                 if id(func) != id(self._map_func_names_to_func_details[func.__name__]['func_obj']):
@@ -429,11 +428,10 @@ def submit(self, func, resource_specification, *args, **kwargs):
         # Get path to files that will contain the pickled function,
         # arguments, result, and map of input and output files
         if exec_mode == 'serverless':
-            if func.__name__ not in self._map_func_names_to_func_details or\
-                    'function_file' not in self._map_func_names_to_func_details[func.__name__]:
+            if 'function_file' not in self._map_func_names_to_func_details[func.__name__]:
                 function_file = os.path.join(self._function_data_dir.name, func.__name__, 'function')
-                self._map_func_names_to_func_details[func.__name__] = {'function_file': function_file, 'is_serialized': False}
                 os.makedirs(os.path.join(self._function_data_dir.name, func.__name__))
+                self._map_func_names_to_func_details[func.__name__].update({'function_file': function_file, 'is_serialized': False})
             else:
                 function_file = self._map_func_names_to_func_details[func.__name__]['function_file']
         else:
@@ -450,7 +448,7 @@ def submit(self, func, resource_specification, *args, **kwargs):
                     function_context_kwargs = resource_specification.get('function_context_kwargs', {})
                     function_context_file = os.path.join(self._function_data_dir.name, func.__name__, 'function_context')
                     self._serialize_object_to_file(function_context_file, [function_context, function_context_args, function_context_kwargs])
-                    self._map_func_names_to_func_details[func.__name__]['function_context_file'] = function_context_file
+                    self._map_func_names_to_func_details[func.__name__].update({'function_context_file': function_context_file})
                 else:
                     function_context_file = self._map_func_names_to_func_details[func.__name__]['function_context_file']