From f1c78138aa28e58eeaafa4791788fe6ceddf1dd8 Mon Sep 17 00:00:00 2001
From: Roger Wang <136131678+ywang96@users.noreply.github.com>
Date: Wed, 3 Jul 2024 00:13:56 -0700
Subject: [PATCH] [Doc] Fix Mock Import (#6094)

---
 docs/source/conf.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/source/conf.py b/docs/source/conf.py
index af1f22b2..7c5bb8f7 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -96,6 +96,7 @@ autodoc_mock_imports = [
     'triton',
     "tqdm",
     "tensorizer",
+    "pynvml",
 ]
 
 for mock_target in autodoc_mock_imports: