From d97191f594129f15accf92e6f4333c7e554c3b2f Mon Sep 17 00:00:00 2001 From: Frederik Hennig <frederik.hennig@fau.de> Date: Fri, 18 Oct 2024 09:31:08 +0200 Subject: [PATCH] fix mypy; reformat all --- src/pystencilssfg/__init__.py | 3 ++- src/pystencilssfg/composer/basic_composer.py | 3 ++- src/pystencilssfg/ir/source_components.py | 4 +--- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/pystencilssfg/__init__.py b/src/pystencilssfg/__init__.py index 1ee4eae..7fa3378 100644 --- a/src/pystencilssfg/__init__.py +++ b/src/pystencilssfg/__init__.py @@ -15,4 +15,5 @@ __all__ = [ ] from . import _version -__version__ = _version.get_versions()['version'] + +__version__ = _version.get_versions()["version"] diff --git a/src/pystencilssfg/composer/basic_composer.py b/src/pystencilssfg/composer/basic_composer.py index 4bdf8b3..1373f1c 100644 --- a/src/pystencilssfg/composer/basic_composer.py +++ b/src/pystencilssfg/composer/basic_composer.py @@ -366,7 +366,7 @@ class SfgBasicComposer(SfgIComposer): def parse_args(*args: ExprLike): args_str = ", ".join(str(arg) for arg in args) - deps = reduce(set.union, (depends(arg) for arg in args), set()) + deps: set[SfgVar] = reduce(set.union, (depends(arg) for arg in args), set()) return SfgStatements( f"{lhs_var.dtype} {lhs_var.name} {{ {args_str} }};", (lhs_var,), @@ -581,6 +581,7 @@ class SfgBranchBuilder(SfgNodeBuilder): class SfgSwitchBuilder(SfgNodeBuilder): """Builder for C++ switches.""" + def __init__(self, switch_arg: ExprLike): self._switch_arg = switch_arg self._cases: dict[str, SfgSequence] = dict() diff --git a/src/pystencilssfg/ir/source_components.py b/src/pystencilssfg/ir/source_components.py index 5007851..8a4e909 100644 --- a/src/pystencilssfg/ir/source_components.py +++ b/src/pystencilssfg/ir/source_components.py @@ -550,9 +550,7 @@ class SfgClass: self, visibility: SfgVisibility | None = None ) -> Generator[SfgMethod, None, None]: if visibility is not None: - yield from filter( - lambda m: m.visibility == visibility, self._methods - ) + yield from filter(lambda m: m.visibility == visibility, self._methods) else: yield from self._methods -- GitLab