From 370f658f6e5e04a4947ec7183e1ed6e0ec2cccf9 Mon Sep 17 00:00:00 2001
From: Michael Kuron <m.kuron@gmx.de>
Date: Fri, 29 Nov 2019 10:53:04 +0100
Subject: [PATCH] adapt to new create_staggered_kernel API

---
 pystencils_walberla/codegen.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/pystencils_walberla/codegen.py b/pystencils_walberla/codegen.py
index 3a0ea97..7cf34be 100644
--- a/pystencils_walberla/codegen.py
+++ b/pystencils_walberla/codegen.py
@@ -34,8 +34,7 @@ def generate_sweep(generation_context, class_name, assignments,
         field_swaps: sequence of field pairs (field, temporary_field). The generated sweep only gets the first field
                      as argument, creating a temporary field internally which is swapped with the first field after
                      each iteration.
-        staggered: set to True to create staggered kernels, in this case assignments has to be a tuple with arguments
-                   to `pystencils.create_staggered_kernel`
+        staggered: set to True to create staggered kernels with `pystencils.create_staggered_kernel`
         varying_parameters: Depending on the configuration, the generated kernels may receive different arguments for
                             different setups. To not have to adapt the C++ application when then parameter change,
                             the varying_parameters sequence can contain parameter names, which are always expected by
@@ -52,7 +51,7 @@ def generate_sweep(generation_context, class_name, assignments,
     if not staggered:
         ast = create_kernel(assignments, **create_kernel_params)
     else:
-        ast = create_staggered_kernel(*assignments, **create_kernel_params)
+        ast = create_staggered_kernel(assignments, **create_kernel_params)
 
     def to_name(f):
         return f.name if isinstance(f, Field) else f
-- 
GitLab