From 8fe3031db48639c97bc05843df706e88862c51e2 Mon Sep 17 00:00:00 2001 From: shingjan Date: Mon, 6 Dec 2021 12:20:35 -0800 Subject: [PATCH] address comments --- python/tvm/script/tir/__init__.pyi | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/script/tir/__init__.pyi b/python/tvm/script/tir/__init__.pyi index db6fc56296ad..ac4ee3018f7c 100644 --- a/python/tvm/script/tir/__init__.pyi +++ b/python/tvm/script/tir/__init__.pyi @@ -370,13 +370,13 @@ def unroll( def thread_binding( begin: Union[PrimExpr, int], end: Union[PrimExpr, int], - thread: str = None, + thread: str, annotations: Optional[Mapping[str, Object]] = None, ) -> Iterable[IterVar]: ... @overload def thread_binding( end: Union[PrimExpr, int], - thread: str = None, + thread: str, annotations: Optional[Mapping[str, Object]] = None, ) -> Iterable[IterVar]: ... @overload