fix(spec): remove stale ModelWorkerBatch annotations

Author: bhaktatejas922

python/sglang/srt/speculative/eagle_worker_v2.py

@@ -126,7 +126,7 @@ def _dsv4_eagle_trace_path() -> Optional[Path]:
     return path / f"dsv4_eagle_trace.rank{rank}.pid{pid}.jsonl"
 
 
-def _dsv4_eagle_batch_payload(batch: Optional[ModelWorkerBatch]) -> Dict[str, Any]:
+def _dsv4_eagle_batch_payload(batch: Optional[Any]) -> Dict[str, Any]:
     if batch is None:
         return {"has_batch": False}
     seq_lens = getattr(batch, "seq_lens", None)
@@ -210,7 +210,7 @@ def _trace_dsv4_eagle_event(
     worker: "EagleWorker",
     event: str,
     *,
-    batch: Optional[ModelWorkerBatch] = None,
+    batch: Optional[Any] = None,
     batch_result: Optional[GenerationBatchResult] = None,
     elapsed_ms: Optional[float] = None,
     **extra: Any,
@@ -821,7 +821,7 @@ def _dsv4_eagle_recompute_payload(
 
 def _dsv4_eagle_handoff_payload(
     *,
-    batch: ModelWorkerBatch,
+    batch: Any,
     batch_result: GenerationBatchResult,
     draft_logits_output: Any,
     select_index: torch.Tensor,