Add more more import mocks

can-sun · can-sun · commit 149ca7049363 · 2023-09-22T17:58:19.000-07:00
diff --git a/doc/conf.py b/doc/conf.py
@@ -109,7 +109,7 @@
 # autosectionlabel
 autosectionlabel_prefix_document = True
 
-autodoc_mock_imports = ['pyspark']
+autodoc_mock_imports = ['pyspark', 'feature_store_pyspark']
 
 def setup(app):
     sys.stdout.write("Generating JumpStart model table...")
