fix mypy

tlc-pack · Jul 14, 2021 · dca4367 · dca4367
1 parent e7b42c0
commit dca4367
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py
@@ -43,7 +43,7 @@ class BlockRV(Object):
     """A random variable that refers to a block"""
 
 
-ExprRV = PrimExpr  #  A random variable that evaluates to an integer
+ExprRV = Union[PrimExpr]  #  A random variable that evaluates to an integer
 
 RAND_VAR_TYPE = Union[ExprRV, BlockRV, LoopRV]  # type: ignore # pylint: disable=invalid-name