Reformat with ruff to reduce vertical space.

docs/user/next/workshop/exercises/helpers.py

@@ -25,11 +25,7 @@
 )
 
 
-def random_mask(
-    sizes,
-    *dims,
-    dtype=None,
-) -> MutableLocatedField:
+def random_mask(sizes, *dims, dtype=None) -> MutableLocatedField:
     arr = np.full(shape=sizes, fill_value=False).flatten()
     arr[: int(arr.size * 0.5)] = True
     np.random.shuffle(arr)

src/gt4py/__about__.py

@@ -19,13 +19,7 @@
 from packaging import version as pkg_version
 
 
-__all__ = [
-    "__author__",
-    "__copyright__",
-    "__license__",
-    "__version__",
-    "__version_info__",
-]
+__all__ = ["__author__", "__copyright__", "__license__", "__version__", "__version_info__"]
 
 
 __author__: Final = "ETH Zurich and individual contributors"

src/gt4py/_core/definitions.py

@@ -143,9 +143,7 @@ def is_positive_integral_type(integral_type: type) -> TypeGuard[Type[PositiveInt
 ]  # TODO(egparedes) figure out if PositiveIntegral can be made to work
 
 
-def is_valid_tensor_shape(
-    value: Sequence[IntegralScalar],
-) -> TypeGuard[TensorShape]:
+def is_valid_tensor_shape(value: Sequence[IntegralScalar]) -> TypeGuard[TensorShape]:
     return isinstance(value, collections.abc.Sequence) and all(
         isinstance(v, numbers.Integral) and v > 0 for v in value
     )

src/gt4py/cartesian/backend/base.py

@@ -276,10 +276,7 @@ def check_options(self, options: gt_definitions.BuildOptions) -> None:
                 stacklevel=2,
             )
 
-    def make_module(
-        self,
-        **kwargs: Any,
-    ) -> Type["StencilObject"]:
+    def make_module(self, **kwargs: Any) -> Type["StencilObject"]:
         build_info = self.builder.options.build_info
         if build_info is not None:
             start_time = time.perf_counter()
@@ -412,10 +409,7 @@ def build_extension_module(
         assert module_name == qualified_pyext_name
 
         self.builder.with_backend_data(
-            {
-                "pyext_module_name": module_name,
-                "pyext_file_path": file_path,
-            }
+            {"pyext_module_name": module_name, "pyext_file_path": file_path}
         )
 
         return module_name, file_path

src/gt4py/cartesian/backend/cuda_backend.py

@@ -89,8 +89,7 @@ def visit_FieldDecl(self, node: cuir.FieldDecl, **kwargs):
             sid_ndim = domain_ndim + data_ndim
             if kwargs["external_arg"]:
                 return "py::object {name}, std::array<gt::int_t,{sid_ndim}> {name}_origin".format(
-                    name=node.name,
-                    sid_ndim=sid_ndim,
+                    name=node.name, sid_ndim=sid_ndim
                 )
             else:
                 return pybuffer_to_sid(
@@ -113,12 +112,7 @@ def visit_Program(self, node: cuir.Program, **kwargs):
         assert "module_name" in kwargs
         entry_params = self.visit(node.params, external_arg=True, **kwargs)
         sid_params = self.visit(node.params, external_arg=False, **kwargs)
-        return self.generic_visit(
-            node,
-            entry_params=entry_params,
-            sid_params=sid_params,
-            **kwargs,
-        )
+        return self.generic_visit(node, entry_params=entry_params, sid_params=sid_params, **kwargs)
 
     Program = bindings_main_template()
 
@@ -157,6 +151,5 @@ def generate(self) -> Type["StencilObject"]:
 
         # Generate and return the Python wrapper class
         return self.make_module(
-            pyext_module_name=pyext_module_name,
-            pyext_file_path=pyext_file_path,
+            pyext_module_name=pyext_module_name, pyext_file_path=pyext_file_path
         )

src/gt4py/cartesian/backend/dace_backend.py

@@ -67,8 +67,7 @@ def _serialize_sdfg(sdfg: dace.SDFG):
 
 def _specialize_transient_strides(sdfg: dace.SDFG, layout_map):
     repldict = replace_strides(
-        [array for array in sdfg.arrays.values() if array.transient],
-        layout_map,
+        [array for array in sdfg.arrays.values() if array.transient], layout_map
     )
     sdfg.replace_dict(repldict)
     for state in sdfg.nodes():
@@ -221,9 +220,7 @@ def _sdfg_add_arrays_and_edges(
             ranges = [
                 (o - max(0, e), o - max(0, e) + s - 1, 1)
                 for o, e, s in zip(
-                    origin,
-                    field_info[name].boundary.lower_indices,
-                    inner_sdfg.arrays[name].shape,
+                    origin, field_info[name].boundary.lower_indices, inner_sdfg.arrays[name].shape
                 )
             ]
             ranges += [(0, d, 1) for d in field_info[name].data_dims]
@@ -786,8 +783,7 @@ def generate(self) -> Type["StencilObject"]:
 
         # Generate and return the Python wrapper class
         return self.make_module(
-            pyext_module_name=pyext_module_name,
-            pyext_file_path=pyext_file_path,
+            pyext_module_name=pyext_module_name, pyext_file_path=pyext_file_path
         )
 
 
@@ -826,10 +822,7 @@ class DaceGPUBackend(BaseDaceBackend):
         ),
     }
     MODULE_GENERATOR_CLASS = DaCeCUDAPyExtModuleGenerator
-    options = {
-        **BaseGTBackend.GT_BACKEND_OPTS,
-        "device_sync": {"versioning": True, "type": bool},
-    }
+    options = {**BaseGTBackend.GT_BACKEND_OPTS, "device_sync": {"versioning": True, "type": bool}}
 
     def generate_extension(self, **kwargs: Any) -> Tuple[str, str]:
         return self.make_extension(stencil_ir=self.builder.gtir, uses_cuda=True)

src/gt4py/cartesian/backend/dace_stencil_object.py

@@ -50,10 +50,7 @@ def add_optional_fields(
         if info.access == AccessKind.NONE and name in kwargs and name not in sdfg.arrays:
             outer_array = kwargs[name]
             sdfg.add_array(
-                name,
-                shape=outer_array.shape,
-                dtype=outer_array.dtype,
-                strides=outer_array.strides,
+                name, shape=outer_array.shape, dtype=outer_array.dtype, strides=outer_array.strides
             )
 
     for name, info in parameter_info.items():
@@ -194,8 +191,7 @@ def normalize_args(
             name: (kwargs[name] if name in kwargs else next(args_iter)) for name in arg_names
         }
         arg_infos = _extract_array_infos(
-            field_args=args_as_kwargs,
-            device=backend_cls.storage_info["device"],
+            field_args=args_as_kwargs, device=backend_cls.storage_info["device"]
         )
 
         origin = DaCeStencilObject._normalize_origins(arg_infos, field_info, origin)

src/gt4py/cartesian/backend/gtc_common.py

@@ -59,15 +59,10 @@ def pybuffer_to_sid(
 
     sid_def = """gt::{as_sid}<{ctype}, {sid_ndim},
         gt::integral_constant<int, {unique_index}>>({name})""".format(
-        name=name,
-        ctype=ctype,
-        unique_index=stride_kind_index,
-        sid_ndim=sid_ndim,
-        as_sid=as_sid,
+        name=name, ctype=ctype, unique_index=stride_kind_index, sid_ndim=sid_ndim, as_sid=as_sid
     )
     sid_def = "gt::sid::shift_sid_origin({sid_def}, {name}_origin)".format(
-        sid_def=sid_def,
-        name=name,
+        sid_def=sid_def, name=name
     )
     if domain_ndim != 3:
         gt_dims = [
@@ -145,10 +140,7 @@ def __init__(self):
         self.pyext_file_path = None
 
     def __call__(
-        self,
-        args_data: ModuleData,
-        builder: Optional["StencilBuilder"] = None,
-        **kwargs: Any,
+        self, args_data: ModuleData, builder: Optional["StencilBuilder"] = None, **kwargs: Any
     ) -> str:
         self.pyext_module_name = kwargs["pyext_module_name"]
         self.pyext_file_path = kwargs["pyext_file_path"]

src/gt4py/cartesian/backend/gtcpp_backend.py

@@ -115,12 +115,7 @@ def visit_Program(self, node: gtcpp.Program, **kwargs):
         assert "module_name" in kwargs
         entry_params = self.visit(node.parameters, external_arg=True, **kwargs)
         sid_params = self.visit(node.parameters, external_arg=False, **kwargs)
-        return self.generic_visit(
-            node,
-            entry_params=entry_params,
-            sid_params=sid_params,
-            **kwargs,
-        )
+        return self.generic_visit(node, entry_params=entry_params, sid_params=sid_params, **kwargs)
 
     Program = bindings_main_template()
 
@@ -151,8 +146,7 @@ def generate(self) -> Type["StencilObject"]:
 
         # Generate and return the Python wrapper class
         return self.make_module(
-            pyext_module_name=pyext_module_name,
-            pyext_file_path=pyext_file_path,
+            pyext_module_name=pyext_module_name, pyext_file_path=pyext_file_path
         )
 
 

src/gt4py/cartesian/backend/module_generator.py

@@ -141,10 +141,7 @@ def __init__(self, builder: Optional["StencilBuilder"] = None):
         )
 
     def __call__(
-        self,
-        args_data: ModuleData,
-        builder: Optional["StencilBuilder"] = None,
-        **kwargs: Any,
+        self, args_data: ModuleData, builder: Optional["StencilBuilder"] = None, **kwargs: Any
     ) -> str:
         """
         Generate source code for a Python module containing a StencilObject.

src/gt4py/cartesian/backend/numpy_backend.py

@@ -117,12 +117,7 @@ def _make_npir(self) -> npir.Computation:
         oir_pipeline = self.builder.options.backend_opts.get(
             "oir_pipeline",
             DefaultPipeline(
-                skip=[
-                    IJCacheDetection,
-                    KCacheDetection,
-                    PruneKCacheFills,
-                    PruneKCacheFlushes,
-                ]
+                skip=[IJCacheDetection, KCacheDetection, PruneKCacheFills, PruneKCacheFlushes]
             ),
         )
         oir_node = oir_pipeline.run(base_oir)

src/gt4py/cartesian/backend/pyext_builder.py

@@ -174,11 +174,7 @@ def get_gt_pyext_build_opts(
 # The following tells mypy to accept unpacking kwargs
 @overload
 def build_pybind_ext(
-    name: str,
-    sources: list,
-    build_path: str,
-    target_path: str,
-    **kwargs: str,
+    name: str, sources: list, build_path: str, target_path: str, **kwargs: str
 ) -> Tuple[str, str]: ...
 
 
@@ -283,11 +279,7 @@ def build_pybind_ext(
 # The following tells mypy to accept unpacking kwargs
 @overload
 def build_pybind_cuda_ext(
-    name: str,
-    sources: list,
-    build_path: str,
-    target_path: str,
-    **kwargs: str,
+    name: str, sources: list, build_path: str, target_path: str, **kwargs: str
 ) -> Tuple[str, str]:
     pass
 

src/gt4py/cartesian/cli.py

@@ -60,10 +60,7 @@ class BackendChoice(click.Choice):
     name = "backend"
 
     def convert(
-        self,
-        value: str,
-        param: Optional[click.Parameter],
-        ctx: Optional[click.Context],
+        self, value: str, param: Optional[click.Parameter], ctx: Optional[click.Context]
     ) -> Type[CLIBackendMixin]:
         """Convert a CLI option argument to a backend."""
         name = super().convert(value, param, ctx)
@@ -239,10 +236,7 @@ def write_computation_src(
                 self.reporter.echo(f"Writing source file: {file_path}")
                 file_path.write_text(content)
 
-    def generate_stencils(
-        self,
-        build_options: Optional[Dict[str, Any]] = None,
-    ) -> None:
+    def generate_stencils(self, build_options: Optional[Dict[str, Any]] = None) -> None:
         for proto_stencil in self.iterate_stencils():
             self.reporter.echo(f"Building stencil {proto_stencil.builder.options.name}")
             builder = proto_stencil.builder.with_backend(self.backend_cls.name)
@@ -314,8 +308,5 @@ def gen(
 ) -> None:
     """Generate stencils from gtscript modules or packages."""
     GTScriptBuilder(
-        input_path=input_path,
-        output_path=output_path,
-        backend=backend,
-        silent=silent,
+        input_path=input_path, output_path=output_path, backend=backend, silent=silent
     ).generate_stencils(build_options=dict(options))

src/gt4py/cartesian/frontend/__init__.py

@@ -16,10 +16,4 @@
 from .base import REGISTRY, Frontend, from_name, register
 
 
-__all__ = [
-    "gtscript_frontend",
-    "REGISTRY",
-    "Frontend",
-    "from_name",
-    "register",
-]
+__all__ = ["gtscript_frontend", "REGISTRY", "Frontend", "from_name", "register"]

src/gt4py/cartesian/frontend/defir_to_gtir.py

@@ -196,10 +196,7 @@ def visit_FieldRef(self, node: FieldRef, *, fields_decls: Dict[str, FieldDecl],
                     data_type = DataType.INT32
                     data_index = [ScalarLiteral(value=index, data_type=data_type)]
                     element_ref = FieldRef(
-                        name=node.name,
-                        offset=node.offset,
-                        data_index=data_index,
-                        loc=node.loc,
+                        name=node.name, offset=node.offset, data_index=data_index, loc=node.loc
                     )
                     field_list.append(element_ref)
             # matrix
@@ -398,9 +395,7 @@ def visit_ComputationBlock(self, node: ComputationBlock) -> gtir.VerticalLoop:
                 stmts.append(decl_or_stmt)
         start, end = self.visit(node.interval)
         interval = gtir.Interval(
-            start=start,
-            end=end,
-            loc=location_to_source_location(node.interval.loc),
+            start=start, end=end, loc=location_to_source_location(node.interval.loc)
         )
         return gtir.VerticalLoop(
             interval=interval,
@@ -525,8 +520,7 @@ def make_bound_or_level(bound: AxisBound, level) -> Optional[common.AxisBound]:
         }
 
         return gtir.HorizontalRestriction(
-            mask=common.HorizontalMask(**axes),
-            body=self.visit(node.body),
+            mask=common.HorizontalMask(**axes), body=self.visit(node.body)
         )
 
     def visit_While(self, node: While) -> gtir.While:

src/gt4py/cartesian/frontend/gtscript_frontend.py

@@ -658,8 +658,7 @@ def _make_init_computations(
         else:
             stmts.append(
                 nodes.Assign(
-                    target=nodes.FieldRef.at_center(name, axes=decl.axes),
-                    value=init_values[name],
+                    target=nodes.FieldRef.at_center(name, axes=decl.axes), value=init_values[name]
                 )
             )
 
@@ -982,8 +981,7 @@ def _visit_computation_node(self, node: ast.With) -> nodes.ComputationBlock:
         if intervals_dicts:
             stmts = [
                 nodes.HorizontalIf(
-                    intervals=intervals_dict,
-                    body=nodes.BlockStmt(stmts=stmts, loc=loc),
+                    intervals=intervals_dict, body=nodes.BlockStmt(stmts=stmts, loc=loc)
                 )
                 for intervals_dict in intervals_dicts
             ]
@@ -1010,9 +1008,7 @@ def visit_Constant(
         elif isinstance(value, bool):
             return nodes.Cast(
                 data_type=nodes.DataType.BOOL,
-                expr=nodes.BuiltinLiteral(
-                    value=nodes.Builtin.from_value(value),
-                ),
+                expr=nodes.BuiltinLiteral(value=nodes.Builtin.from_value(value)),
                 loc=nodes.Location.from_ast_node(node),
             )
         elif isinstance(value, numbers.Number):

src/gt4py/cartesian/gtc/common.py

@@ -406,10 +406,7 @@ def _make_root_validator(impl: datamodels.RootValidator) -> datamodels.RootValid
 
 
 def assign_stmt_dtype_validation(*, strict: bool) -> datamodels.RootValidator:
-    def _impl(
-        cls: Type[datamodels.DataModel],
-        instance: datamodels.DataModel,
-    ) -> None:
+    def _impl(cls: Type[datamodels.DataModel], instance: datamodels.DataModel) -> None:
         assert isinstance(instance, AssignStmt)
         verify_and_get_common_dtype(cls, [instance.left, instance.right], strict=strict)
 
@@ -865,10 +862,7 @@ def data_type_to_typestr(dtype: DataType) -> str:
         ComparisonOperator.EQ: "equal",
         ComparisonOperator.NE: "not_equal",
     },
-    LogicalOperator: {
-        LogicalOperator.AND: "logical_and",
-        LogicalOperator.OR: "logical_or",
-    },
+    LogicalOperator: {LogicalOperator.AND: "logical_and", LogicalOperator.OR: "logical_or"},
     NativeFunction: {
         NativeFunction.ABS: "abs",
         NativeFunction.MIN: "minimum",