fix docs build script

google-gemini · Mar 25, 2024 · 4243783 · 4243783
1 parent 475bafd
commit 4243783
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/build_docs.py b/docs/build_docs.py
@@ -36,7 +36,8 @@
 from google.ai import generativelanguage as glm
 
 import grpc
-import pydantic #must be imported before turning on TYPE_CHECKING
+import pydantic  # must be imported before turning on TYPE_CHECKING
+from IPython import display   # must be imported before turning on TYPE_CHECKING
 
 # For showing the conditional imports and types in `content_types.py`
 # grpc must be imported first.
@@ -165,7 +166,6 @@ def make_default_filters(self):
             public_api.FailIfNestedTooDeep(10),
             public_api.filter_module_all,
             public_api.add_proto_fields,
-            public_api.filter_builtin_modules,
             public_api.filter_private_symbols,
             MyFilter(self._base_dir),  # Replaces: public_api.FilterBaseDirs(self._base_dir),
             public_api.FilterPrivateMap(self._private_map),