add .pre-commit-config.yaml and docs/conf.py to owlbot excludes

Author: dizcology

.pre-commit-config.yaml

@@ -26,6 +26,6 @@ repos:
     hooks:
     - id: black
 -   repo: https://github.com/pycqa/flake8
-    rev: 3.9.2
+    rev: 6.1.0
     hooks:
     - id: flake8

docs/conf.py

@@ -382,3 +382,57 @@
 napoleon_use_ivar = False
 napoleon_use_param = True
 napoleon_use_rtype = True
+
+
+def adopt_members_reexported_from_private_modules(public_module: str):
+    """Remaps the module items that come from internal modules.
+
+    A public module might be exporting items that are imported from private modules.
+    This function changes the `__module__` of such items to the public module.
+
+    Example:
+        `package/public.py`:
+
+        ```
+        from package._private import _PrivateClass as PublicClass
+        __all__ = ["PublicClass"]
+        ```
+
+        Calling this function on the `package.public` module will change:
+        ```
+        package._private._PrivateClass.__name__ = "PublicClass"
+        package._private._PrivateClass.__module__ = "package.public"
+        ```
+    """
+    for name, cls in public_module.__dict__.items():
+        if name in public_module.__all__:
+            if "._" in cls.__module__:
+                cls.__name__ = name
+                cls.__module__ = public_module.__name__
+
+
+def setup(*args, **kwargs):
+    # 1. Giving pretty module names to the GA and preview classes
+    # 2. Giving pretty class names to the preview classes
+    # 3. Making Sphinx automodule render the class members instead of
+    #     dismissing the exported private classes as "Alias of".
+    from vertexai import language_models
+    from vertexai import vision_models
+    from vertexai.preview import (
+        language_models as preview_language_models,
+    )
+    from vertexai.preview import (
+        vision_models as preview_vision_models,
+    )
+
+    # There are many possible ways to select which classes to fix.
+    # We select the publicly exported members that have an internal module ("*._*").
+
+    # Setting the modules of the GA classes
+    adopt_members_reexported_from_private_modules(language_models)
+    adopt_members_reexported_from_private_modules(vision_models)
+
+    # Setting the modules of the public preview classes
+    # Selecting the members that still have an internal module after the GA fixes.
+    adopt_members_reexported_from_private_modules(preview_language_models)
+    adopt_members_reexported_from_private_modules(preview_vision_models)

owlbot.py

@@ -82,6 +82,8 @@
             ".kokoro/samples",
             "noxfile.py",
             "testing",
+            ".pre-commit-config.yaml",
+            "docs/conf.py",
         ],
     )
     has_generator_updates = True