3
3
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
4
4
5
5
from typing import (
6
+ Any as _Any ,
6
7
List as _List ,
7
8
Optional as _Optional ,
8
9
Sequence as _Sequence ,
12
13
)
13
14
14
15
from .._mlir_libs import _mlir as _cext
15
- from ..ir import (
16
+ from ..ir import ( # type: ignore
16
17
ArrayAttr ,
17
18
Attribute ,
18
19
BoolAttr ,
@@ -143,7 +144,7 @@ def get_op_result_or_op_results(
143
144
else op
144
145
)
145
146
146
- ResultValueTypeTuple = _cext .ir .Operation , _cext .ir .OpView , _cext .ir .Value
147
+ ResultValueTypeTuple = _Tuple [ _cext .ir .Operation , _cext .ir .OpView , _cext .ir .Value ]
147
148
ResultValueT = _Union [ResultValueTypeTuple ]
148
149
VariadicResultValueT = _Union [ResultValueT , _Sequence [ResultValueT ]]
149
150
@@ -244,8 +245,8 @@ def _dispatch_mixed_values(
244
245
245
246
246
247
def _get_value_or_attribute_value (
247
- value_or_attr : _Union [any , Attribute , ArrayAttr ]
248
- ) -> any :
248
+ value_or_attr : _Union [_Any , Attribute , ArrayAttr ]
249
+ ) -> _Any :
249
250
if isinstance (value_or_attr , Attribute ) and hasattr (value_or_attr , "value" ):
250
251
return value_or_attr .value
251
252
if isinstance (value_or_attr , ArrayAttr ):
@@ -254,8 +255,8 @@ def _get_value_or_attribute_value(
254
255
255
256
256
257
def _get_value_list (
257
- sequence_or_array_attr : _Union [_Sequence [any ], ArrayAttr ]
258
- ) -> _Sequence [any ]:
258
+ sequence_or_array_attr : _Union [_Sequence [_Any ], ArrayAttr ]
259
+ ) -> _Sequence [_Any ]:
259
260
return [_get_value_or_attribute_value (v ) for v in sequence_or_array_attr ]
260
261
261
262
0 commit comments