diff --git a/src/pystencilssfg/__init__.py b/src/pystencilssfg/__init__.py index 1ee4eaee9df05fc447c990bde3a889066675d850..7fa3378c7958c13f4fb526eb67a2bd08335400f5 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 4bdf8b30cc8cf7043d862b72af8d5f2b1f27c227..1373f1cb43e9609fee7d34be5c1ca1abb5785b66 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 5007851eeb9027ef025c5e46c3d6da4c019cf2c0..8a4e90967d74aabc4f50529c7ff0e51c76a3d869 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