Add ArithArrayContainerT

Author: inducer

arraycontext/__init__.py

@@ -88,6 +88,7 @@
 from .transform_metadata import CommonSubexpressionTag, ElementwiseMapKernelTag
 from .typing import (
     ArithArrayContainer,
+    ArithArrayContainerT,
     Array,
     ArrayContainer,
     ArrayContainerT,
@@ -110,6 +111,7 @@
 
 __all__ = (
     "ArithArrayContainer",
+    "ArithArrayContainerT",
     "Array",
     "ArrayContainer",
     "ArrayContainerT",

arraycontext/container/__init__.py

@@ -50,6 +50,7 @@
 
 .. autoclass:: ArithArrayContainer
 .. autoclass:: ArrayContainerT
+.. autoclass:: ArithArrayContainerT
 
 .. autoexception:: NotAnArrayContainerError
 

arraycontext/typing.py

@@ -231,6 +231,7 @@ def __rpow__(self, other: ArrayOrScalar | Self) -> Self: ...
 
 
 ArrayContainerT = TypeVar("ArrayContainerT", bound=ArrayContainer)
+ArithArrayContainerT = TypeVar("ArithArrayContainerT", bound=ArithArrayContainer)
 
 # }}}