From 31ae4a13a328bc2175174fb3dc1829a8a88f7bc6 Mon Sep 17 00:00:00 2001
From: rtmadduri <170052006+rtmadduri@users.noreply.github.com>
Date: Tue, 17 Feb 2026 22:17:07 +0000
Subject: [PATCH] chore: Update coverage include list for AMD-modified files

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3d3ce247fc..82d1a01dc2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -204,6 +204,7 @@ branch = true
 # AMD/HIP modified files - auto-updated by GitHub Action - DO NOT EDIT THIS SECTION MANUALLY
 include = [
     "flashinfer/__init__.py",
+    "flashinfer/aiter_utils.py",
     "flashinfer/aot_hip.py",
     "flashinfer/compilation_context_hip.py",
     "flashinfer/decode_rocm.py",
@@ -222,7 +223,6 @@ include = [
     "flashinfer/prefill_rocm.py",
     "flashinfer/quantization.py",
     "flashinfer/rope.py",
-    "flashinfer/sampling.py",
     "flashinfer/utils.py",
 ]