ExpressionT

inducer · Jan 13, 2025 · 9b12a27 · 9b12a27
1 parent c67eea6
commit 9b12a27
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/loopy/statistics.py b/loopy/statistics.py
@@ -69,13 +69,13 @@
     from collections.abc import Mapping, Sequence
 
     import pymbolic.primitives as p
-    from pymbolic.typing import ArithmeticExpressionT
+    from pymbolic.typing import ArithmeticExpressionT, ExpressionT
     from pytools.tag import Tag
 
     from loopy.kernel.array import ArrayBase
     from loopy.kernel.instruction import InstructionBase
     from loopy.types import LoopyType
-    from loopy.typing import Expression, ExpressionT, auto
+    from loopy.typing import Expression, auto
 
 
 __doc__ = """

diff --git a/loopy/symbolic.py b/loopy/symbolic.py
@@ -79,6 +79,7 @@
 )
 from pymbolic.mapper.unifier import UnidirectionalUnifier as UnidirectionalUnifierBase
 from pymbolic.parser import Parser as ParserBase
+from pymbolic.typing import ExpressionT
 from pytools import memoize, memoize_method, memoize_on_first_arg
 from pytools.tag import Tag, Taggable, ToTagSetConvertible
 
@@ -87,7 +88,7 @@
     LoopyError,
     UnableToDetermineAccessRangeError,
 )
-from loopy.typing import Expression, ExpressionT, not_none
+from loopy.typing import Expression, not_none
 
 
 if TYPE_CHECKING: