From 50ac280ae7028ddda73ef6107dcb6bae23c5815f Mon Sep 17 00:00:00 2001
From: Dmytro Dzhulgakov <dima.v.dzhulgakov@gmail.com>
Date: Thu, 13 Mar 2025 00:42:08 +0000
Subject: [PATCH] comments

---
 deep_ep/buffer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deep_ep/buffer.py b/deep_ep/buffer.py
index 2d7a0e6..098b346 100644
--- a/deep_ep/buffer.py
+++ b/deep_ep/buffer.py
@@ -498,7 +498,7 @@ class Buffer:
     # noinspection PyTypeChecker
     def low_latency_combine(self, x: torch.Tensor, topk_idx: torch.Tensor, topk_weights: torch.Tensor,
                             handle: tuple, async_finish: bool = False, return_recv_hook: bool = False,
-                            out: torch.Tensor | None = None) -> \
+                            out: Optional[torch.Tensor] = None) -> \
             Tuple[torch.Tensor, EventOverlap, Callable]:
         """
         A low-latency implementation for combining tokens (reduce **with weights**) with IBGDA.