Skip to content

Commit

Permalink
#2410 styling changes from review
Browse files Browse the repository at this point in the history
  • Loading branch information
mo-lottieturner committed Jan 22, 2024
1 parent 3bda21f commit ed1ea8b
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 6 deletions.
4 changes: 2 additions & 2 deletions src/psyclone/domain/lfric/kern_call_acc_arg_list.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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):
Expand Down
3 changes: 2 additions & 1 deletion src/psyclone/domain/lfric/lfric_kern.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
1 change: 1 addition & 0 deletions src/psyclone/domain/lfric/lfric_stencils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions src/psyclone/tests/domain/lfric/lfric_stencil_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit ed1ea8b

Please sign in to comment.