address comments

apache · Dec 7, 2021 · 8fe3031 · 8fe3031
1 parent e27d05f
commit 8fe3031
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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