Update schedule.py

Author: vinx13

python/tvm/tir/schedule/schedule.py

@@ -1664,6 +1664,7 @@ def after_storage_align(a: T.handle, c: T.handle) -> None:
 
     ########## Schedule: Annotation ##########
 
+    @type_checked
     def annotate(
         self,
         block_or_loop: Union[BlockRV, LoopRV],
@@ -1730,6 +1731,7 @@ def after_annotate(a: T.handle, b: T.handle) -> None:
             self, block_or_loop, ann_key, ann_val
         )
 
+    @type_checked
     def unannotate(self, block_or_loop: Union[BlockRV, LoopRV], ann_key: str) -> None:
         """Unannotate a block/loop's annotation with key ann_key