diff --git a/pykokkos/core/parsers/parser.py b/pykokkos/core/parsers/parser.py index 55a2dfe6..f2913c61 100644 --- a/pykokkos/core/parsers/parser.py +++ b/pykokkos/core/parsers/parser.py @@ -369,14 +369,12 @@ def fix_decorator( # no change needed return entity_tree.decorator_list - call_obj = ast.Call() - call_obj.func = ast.Attribute( + func = ast.Attribute( value=ast.Name(id=self.pk_import, ctx=ast.Load()), attr="workunit", ctx=ast.Load(), ) - call_obj.args = [] - call_obj.keywords = [] + call_obj = ast.Call(func=func, args=[], keywords=[]) for view, specifier_dict in updated_decorator.inferred_decorator.items(): call_obj.keywords.append(self.get_keyword_node(view, specifier_dict))