From a9050b5c39dee55730382b73fc03735b029cc65e Mon Sep 17 00:00:00 2001
From: wejoncy <247153481@qq.com>
Date: Sun, 28 Sep 2025 16:26:20 +0800
Subject: [PATCH] [bugfix]Add empty_context import to two_batch_overlap.py
 (#10964)

---
 python/sglang/srt/two_batch_overlap.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sglang/srt/two_batch_overlap.py b/python/sglang/srt/two_batch_overlap.py
index e02bc1fd2..82717b382 100644
--- a/python/sglang/srt/two_batch_overlap.py
+++ b/python/sglang/srt/two_batch_overlap.py
@@ -31,7 +31,7 @@ from sglang.srt.model_executor.forward_batch_info import (
 from sglang.srt.operations import execute_operations, execute_overlapped_operations
 from sglang.srt.operations_strategy import OperationsStrategy
 from sglang.srt.speculative.eagle_utils import EagleDraftInput, EagleVerifyInput
-from sglang.srt.utils import BumpAllocator, get_bool_env_var, is_hip
+from sglang.srt.utils import BumpAllocator, empty_context, get_bool_env_var, is_hip
 
 if TYPE_CHECKING:
     from sglang.srt.layers.moe.token_dispatcher import DispatchOutput