We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bb04839 commit 18dc886Copy full SHA for 18dc886
1 file changed
onnxscript/type_annotation.py
@@ -131,7 +131,7 @@ def _is_tensor_type(typeinfo: TypeAnnotationValue) -> bool:
131
if type(typeinfo) is onnx_types.TensorType:
132
# Special case the handle when typeinfo is TensorType.
133
# It seems abc.ABC in py39 has issues with issubclass
134
- return False
+ return True
135
if inspect.isclass(typeinfo) and issubclass(typeinfo, onnx_types.TensorType):
136
return True
137
return False
0 commit comments