From f9bcd490abdd8054349c4fabc83d539911b46c88 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 4c3b7226a6f1..7b7213526778 100644 --- a/python/tvm/script/tir/__init__.pyi +++ b/python/tvm/script/tir/__init__.pyi @@ -364,13 +364,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