From dd164d72f31f95e19d84756a644811d0ca2c7c59 Mon Sep 17 00:00:00 2001
From: Cyrus Leung <tlleungac@connect.ust.hk>
Date: Wed, 14 Aug 2024 11:37:30 +0800
Subject: [PATCH] [Bugfix][Docs] Update list of mock imports (#7493)

---
 .buildkite/test-pipeline.yaml | 2 ++
 docs/source/conf.py           | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml
index 6f226f0f3101..46e259116691 100644
--- a/.buildkite/test-pipeline.yaml
+++ b/.buildkite/test-pipeline.yaml
@@ -36,6 +36,8 @@ steps:
   commands:
   - pip install -r requirements-docs.txt
   - SPHINXOPTS=\"-W\" make html
+  # Check API reference (if it fails, you may have missing mock imports)
+  - grep \"sig sig-object py\" build/html/dev/sampling_params.html
 
 - label: Async Engine, Inputs, Utils, Worker Test # 15min
   fast_check: true
diff --git a/docs/source/conf.py b/docs/source/conf.py
index d24ed0320782..b4f5b4ab9d56 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -97,13 +97,13 @@ def setup(app):
 # Mock out external dependencies here, otherwise the autodoc pages may be blank.
 autodoc_mock_imports = [
     "aiohttp",
+    "compressed_tensors",
     "cpuinfo",
     "torch",
     "transformers",
     "psutil",
     "prometheus_client",
     "sentencepiece",
-    "vllm.cuda_utils",
     "vllm._C",
     "PIL",
     "numpy",