Skip to content
Snippets Groups Projects
Commit 55eb1aa0 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Merge branch 'StrictUndefined' into 'master'

Use StrictUndefined for creating jinja2.Environment

See merge request !7
parents c234592a e619403e
Branches
Tags
1 merge request!7Use StrictUndefined for creating jinja2.Environment
Pipeline #21910 passed with stage
in 38 seconds
......@@ -2,7 +2,7 @@ from collections import OrderedDict, defaultdict
from itertools import product
from typing import Dict, Optional, Sequence, Tuple
from jinja2 import Environment, PackageLoader
from jinja2 import Environment, PackageLoader, StrictUndefined
from pystencils import (
Assignment, AssignmentCollection, Field, FieldType, create_kernel, create_staggered_kernel)
......@@ -61,7 +61,7 @@ def generate_sweep(generation_context, class_name, assignments,
ast.function_name = class_name.lower()
env = Environment(loader=PackageLoader('pystencils_walberla'))
env = Environment(loader=PackageLoader('pystencils_walberla'), undefined=StrictUndefined)
add_pystencils_filters_to_jinja_env(env)
if inner_outer_split is False:
......@@ -240,7 +240,7 @@ def generate_pack_info(generation_context, class_name: str,
'field_name': field_names.pop(),
'namespace': namespace,
}
env = Environment(loader=PackageLoader('pystencils_walberla'))
env = Environment(loader=PackageLoader('pystencils_walberla'), undefined=StrictUndefined)
add_pystencils_filters_to_jinja_env(env)
header = env.get_template(template_name + ".h").render(**jinja_context)
source = env.get_template(template_name + ".cpp").render(**jinja_context)
......@@ -301,7 +301,7 @@ def generate_mpidtype_info_from_kernel(generation_context, class_name: str,
'f_size': field.index_shape[0],
'spec': spec,
}
env = Environment(loader=PackageLoader('pystencils_walberla'))
env = Environment(loader=PackageLoader('pystencils_walberla'), undefined=StrictUndefined)
header = env.get_template("MpiDtypeInfo.tmpl.h").render(**jinja_context)
generation_context.write_file("{}.h".format(class_name), header)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment