fix: sphinx try 2

AstraZeneca · Sep 18, 2024 · eb9c994 · eb9c994
1 parent 2ebbbd8
commit eb9c994
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -262,5 +262,7 @@ def linkcode_resolve(domain, info):
     # no sphinx for streamlit
     ("py:class", "streamlit.delta_generator.DeltaGenerator"),
     # vertexai
-    ("py:class", "vertexai.*"),
+    ("py:class", "vertexai.generative_models._generative_models.SafetySetting"),
+    ("py:class", "google.cloud.aiplatform_v1beta1.types.content.HarmCategory"),
+    ("py:class", "google.cloud.aiplatform_v1beta1.types.content.SafetySetting.HarmBlockThreshold"),
 ]