From 1058d1ee1f241dc9b5d0bd44d62fc0fdbaec9abd Mon Sep 17 00:00:00 2001
From: Zihao Ye <expye@outlook.com>
Date: Fri, 8 Nov 2024 13:24:26 -0800
Subject: [PATCH] hotfix: fix import issue in #597 (#598)

#597 didn't import `Union`.
---
 python/flashinfer/page.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/flashinfer/page.py b/python/flashinfer/page.py
index 8efe2168..581d6207 100644
--- a/python/flashinfer/page.py
+++ b/python/flashinfer/page.py
@@ -14,7 +14,7 @@
 limitations under the License.
 """
 
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 import torch
 import triton