diff --git a/python/tvm/script/printer/ir_docsifier.py b/python/tvm/script/printer/ir_docsifier.py index eb9786373648d..16f3ab62ecabd 100644 --- a/python/tvm/script/printer/ir_docsifier.py +++ b/python/tvm/script/printer/ir_docsifier.py @@ -114,9 +114,9 @@ def set_dispatch( raise TypeError(f"{type(node_type)} is not a registered TVM object type.") _ensure_cleanup_function_registered() - _ffi_api.IRDocsifierSetDispatch( # type: ignore + _ffi_api.IRDocsifierSetDispatch( # type: ignore # pylint: disable=no-member dispatch_token, type_index, dispatch_function - ) # pylint: disable=no-member + ) _REGISTERED_TYPES.add((dispatch_token, type_index)) def as_doc(self, obj: Object, object_path: ObjectPath) -> Doc: