Un-type symbolic.flatten

This restores compatibility with pymbolic 2022.2

x-ref: firedrakeproject/loopy#27

loopy/symbolic.py

@@ -37,8 +37,6 @@
     Mapping,
     Sequence,
     Tuple,
-    TypeVar,
-    cast,
 )
 
 import immutables
@@ -49,7 +47,6 @@
 import pymbolic.primitives as p
 import pytools.lex
 from islpy import dim_type
-from pymbolic import ArithmeticExpressionT
 from pymbolic.mapper import (
     CachedCombineMapper as CombineMapperBase,
     CachedIdentityMapper as IdentityMapperBase,
@@ -211,14 +208,8 @@ class FlattenMapper(FlattenMapperBase, IdentityMapperMixin):
     pass
 
 
-ArithmeticOrExpressionT = TypeVar(
-                "ArithmeticOrExpressionT",
-                ArithmeticExpressionT,
-                ExpressionT)
-
-
-def flatten(expr: ArithmeticOrExpressionT) -> ArithmeticOrExpressionT:
-    return cast(ArithmeticOrExpressionT, FlattenMapper()(expr))
+def flatten(expr):
+    return FlattenMapper()(expr)
 
 
 class IdentityMapper(IdentityMapperBase, IdentityMapperMixin):