Placate ruff 0.16.5

Author: inducer

arraycontext/container/arithmetic.py

@@ -738,7 +738,7 @@ def {fname}(arg2, arg1):
         code = gen.get().rstrip()+"\n"
 
         result_dict = {"_MODULE_SOURCE_CODE": code, "cls": cls}
-        exec(compile(code, f"<container arithmetic for {cls.__name__}>", "exec"),  # ruff:ignore[exec-builtin]
+        exec(compile(code, f"<container arithmetic for {cls.__name__}>", "exec"),
                 result_dict)
 
         return cls

arraycontext/container/dataclass.py

@@ -247,7 +247,7 @@ def _deserialize_init_arrays_code_{lower_cls_name}(
         """)
 
     exec_dict = {"cls": cls, "_MODULE_SOURCE_CODE": serialize_code}
-    exec(compile(serialize_code, f"<container serialization for {cls.__name__}>",  # ruff:ignore[exec-builtin]
+    exec(compile(serialize_code, f"<container serialization for {cls.__name__}>",
         "exec"), exec_dict)
 
     return cls

arraycontext/container/traversal.py

@@ -1209,7 +1209,7 @@ def treat_as_scalar(x: ArrayOrContainerOrScalar) -> bool:
             raise TypeError(
                 "both arguments must have the same type if they are both "
                 "non-object-array array containers.")
-        return multimap_array_container(lambda x, y: outer(x, y), a, b)
+        return multimap_array_container(outer, a, b)
 
 # }}}
 

arraycontext/context.py

@@ -495,15 +495,11 @@ def _get_csr_matmul_prg(self, out_ndim: int) -> loopy.TranslationUnit:
         out_shape_comp_names = ("nrows", *out_extra_shape_comp_names)
 
         domains: list[str] = []
-        domains.append(
-            "{ [" + ",".join(out_inames) + "] : "
-            + " and ".join(
-                f"0 <= {iname} < {shape_comp_name}"
-                for iname, shape_comp_name in zip(
-                    out_inames, out_shape_comp_names, strict=True))
-            + " }")
-        domains.append(
-            "{ [iel] : iel_lbound <= iel < iel_ubound }")
+        domains.extend(("{ [" + ",".join(out_inames) + "] : " + " and ".join(
+            f"0 <= {iname} < {shape_comp_name}"
+            for iname, shape_comp_name
+            in zip(out_inames, out_shape_comp_names, strict=True))
+        + " }", "{ [iel] : iel_lbound <= iel < iel_ubound }"))
 
         temporary_variables: Mapping[str, lp.TemporaryVariable] = {
             "iel_lbound": lp.TemporaryVariable(

arraycontext/impl/numpy/fake_numpy.py

@@ -239,11 +239,11 @@ def vdot(self, a, b):
 
     def any(self, a, /):
         return rec_map_reduce_array_container(partial(reduce, np.logical_or),
-                                              lambda subary: np.any(subary), a)
+                                              np.any, a)
 
     def all(self, a, /):
         return rec_map_reduce_array_container(partial(reduce, np.logical_and),
-                                              lambda subary: np.all(subary), a)
+                                              np.all, a)
 
     @override
     def array_equal(self,

arraycontext/impl/pyopencl/__init__.py

@@ -152,7 +152,7 @@ def __init__(self,
         super().__init__()
         self.context = queue.context
         self.queue = queue
-        self.allocator = allocator if allocator else None
+        self.allocator = allocator or None
         if wait_event_queue_length is None:
             wait_event_queue_length = 10
 

arraycontext/impl/pyopencl/fake_numpy.py

@@ -365,7 +365,7 @@ def sum(self,
                 dtype: DTypeLike | None = None,
             ) -> ArrayOrScalar:
         if isinstance(axis, int):
-            axis = axis,
+            axis = (axis,)
 
         def _rec_sum(ary):
             if axis not in [None, tuple(range(ary.ndim))]:
@@ -400,7 +400,7 @@ def max(self,
         queue = self._array_context.queue
 
         if isinstance(axis, int):
-            axis = axis,
+            axis = (axis,)
 
         def _rec_max(ary):
             if axis not in [None, tuple(range(ary.ndim))]:
@@ -438,7 +438,7 @@ def min(self,
         queue = self._array_context.queue
 
         if isinstance(axis, int):
-            axis = axis,
+            axis = (axis,)
 
         def _rec_min(ary):
             if axis not in [None, tuple(range(ary.ndim))]:

arraycontext/impl/pytato/compile.py

@@ -173,7 +173,7 @@ def id_collector(keys, ary):
 def _to_input_for_compiled(
         ary: AllowedArrayTc,
         actx: _BasePytatoArrayContext) -> AllowedArrayTc:
-    """
+    r"""
     Preprocess *ary* before turning it into a :class:`pytato.array.Placeholder`
     in :meth:`LazilyCompilingFunctionCaller.__call__`.
 

arraycontext/impl/pytato/fake_numpy.py

@@ -175,12 +175,12 @@ def stack(self, arrays, axis=0):
     def all(self, a, /):
         return rec_map_reduce_array_container(
                 partial(reduce, pt.logical_and),
-                lambda subary: pt.all(subary), a)
+                pt.all, a)
 
     def any(self, a, /):
         return rec_map_reduce_array_container(
                 partial(reduce, pt.logical_or),
-                lambda subary: pt.any(subary), a)
+                pt.any, a)
 
     @override
     def array_equal(self,

arraycontext/pytest.py

@@ -421,7 +421,7 @@ def inner(metafunc: pytest.Metafunc) -> None:
         arg_value_tuples = sorted([
                 tuple(arg_dict[name] for name in arg_names)
                 for arg_dict in arg_values_with_actx
-                ], key=lambda x: str(x))
+                ], key=str)
 
         metafunc.parametrize(arg_names, arg_value_tuples, ids=ids)