diff --git a/src/psyclone/domain/lfric/kern_call_acc_arg_list.py b/src/psyclone/domain/lfric/kern_call_acc_arg_list.py index 495602722d..564e1b70e6 100644 --- a/src/psyclone/domain/lfric/kern_call_acc_arg_list.py +++ b/src/psyclone/domain/lfric/kern_call_acc_arg_list.py @@ -111,7 +111,7 @@ def stencil(self, arg, var_accesses=None): # pylint: disable=import-outside-toplevel from psyclone.domain.lfric.lfric_stencils import LFRicStencils var_name = LFRicStencils.dofmap_symbol(self._kern.root.symbol_table, - arg).name + arg).name self.append(var_name, var_accesses) def stencil_2d(self, arg, var_accesses=None): @@ -149,7 +149,7 @@ def stencil_unknown_extent(self, arg, var_accesses=None): # pylint: disable=import-outside-toplevel from psyclone.domain.lfric.lfric_stencils import LFRicStencils name = LFRicStencils.dofmap_size_symbol(self._kern.root.symbol_table, - arg).name + arg).name self.append(name, var_accesses) def stencil_2d_unknown_extent(self, arg, var_accesses=None): diff --git a/src/psyclone/domain/lfric/lfric_kern.py b/src/psyclone/domain/lfric/lfric_kern.py index 21475b9734..2f1fd28cab 100644 --- a/src/psyclone/domain/lfric/lfric_kern.py +++ b/src/psyclone/domain/lfric/lfric_kern.py @@ -609,7 +609,8 @@ def gen_stub(self): # Add all the declarations # Import here to avoid circular dependency # pylint: disable=import-outside-toplevel - from psyclone.domain.lfric import LFRicFields, LFRicScalarArgs, LFRicStencils + from psyclone.domain.lfric import (LFRicFields, LFRicScalarArgs, + LFRicStencils) from psyclone.dynamo0p3 import (DynCellIterators, DynDofmaps, DynFunctionSpaces, DynCMAOperators, DynBoundaryConditions, diff --git a/src/psyclone/domain/lfric/lfric_stencils.py b/src/psyclone/domain/lfric/lfric_stencils.py index 566c312565..5651f35fe7 100644 --- a/src/psyclone/domain/lfric/lfric_stencils.py +++ b/src/psyclone/domain/lfric/lfric_stencils.py @@ -51,6 +51,7 @@ IfThenGen, TypeDeclGen, UseGen) from psyclone.psyir.symbols import ScalarType + class LFRicStencils(LFRicCollection): ''' Stencil information and code generation associated with a PSy-layer diff --git a/src/psyclone/tests/domain/lfric/lfric_stencil_test.py b/src/psyclone/tests/domain/lfric/lfric_stencil_test.py index 1f4062f9bc..8dfadb4a48 100644 --- a/src/psyclone/tests/domain/lfric/lfric_stencil_test.py +++ b/src/psyclone/tests/domain/lfric/lfric_stencil_test.py @@ -1976,9 +1976,9 @@ def test_dynstencils_extent_vars_err(monkeypatch): def test_dynstencils_err(): - ''' Check that LFRicStencils.initialise and LFRicStencils._declare_maps_invoke - raises the expected InternalError if an unsupported stencil type is - encountered. ''' + ''' Check that LFRicStencils.initialise and + LFRicStencils._declare_maps_invoke raises the expected + InternalError if an unsupported stencil type is encountered. ''' _, info = parse(os.path.join(BASE_PATH, "19.1_single_stencil.f90"), api=TEST_API) psy = PSyFactory(TEST_API, distributed_memory=True).create(info)