diff --git a/docs/aiplatform/definition_v1/types.rst b/docs/aiplatform/definition_v1.rst
similarity index 100%
rename from docs/aiplatform/definition_v1/types.rst
rename to docs/aiplatform/definition_v1.rst
diff --git a/docs/aiplatform/definition_v1beta1/types.rst b/docs/aiplatform/definition_v1beta1.rst
similarity index 100%
rename from docs/aiplatform/definition_v1beta1/types.rst
rename to docs/aiplatform/definition_v1beta1.rst
diff --git a/docs/aiplatform/instance_v1/types.rst b/docs/aiplatform/instance_v1.rst
similarity index 100%
rename from docs/aiplatform/instance_v1/types.rst
rename to docs/aiplatform/instance_v1.rst
diff --git a/docs/aiplatform/instance_v1beta1/types.rst b/docs/aiplatform/instance_v1beta1.rst
similarity index 100%
rename from docs/aiplatform/instance_v1beta1/types.rst
rename to docs/aiplatform/instance_v1beta1.rst
diff --git a/docs/aiplatform/params_v1/types.rst b/docs/aiplatform/params_v1.rst
similarity index 100%
rename from docs/aiplatform/params_v1/types.rst
rename to docs/aiplatform/params_v1.rst
diff --git a/docs/aiplatform/params_v1beta1/types.rst b/docs/aiplatform/params_v1beta1.rst
similarity index 100%
rename from docs/aiplatform/params_v1beta1/types.rst
rename to docs/aiplatform/params_v1beta1.rst
diff --git a/docs/aiplatform/prediction_v1/types.rst b/docs/aiplatform/prediction_v1.rst
similarity index 100%
rename from docs/aiplatform/prediction_v1/types.rst
rename to docs/aiplatform/prediction_v1.rst
diff --git a/docs/aiplatform/prediction_v1beta1/types.rst b/docs/aiplatform/prediction_v1beta1.rst
similarity index 100%
rename from docs/aiplatform/prediction_v1beta1/types.rst
rename to docs/aiplatform/prediction_v1beta1.rst
diff --git a/noxfile.py b/noxfile.py
index b2809b5073..2e7154fd4e 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -280,7 +280,12 @@ def docs(session):
     """Build the docs for this library."""
 
     session.install("-e", ".")
-    session.install("sphinx==4.0.1", "alabaster", "recommonmark")
+    session.install(
+        "sphinx==4.0.1",
+        "alabaster",
+        "google-cloud-aiplatform[prediction]",
+        "recommonmark",
+    )
 
     shutil.rmtree(os.path.join("docs", "_build"), ignore_errors=True)
     session.run(
@@ -302,7 +307,11 @@ def docfx(session):
 
     session.install("-e", ".")
     session.install(
-        "sphinx==4.0.1", "alabaster", "recommonmark", "gcp-sphinx-docfx-yaml"
+        "sphinx==4.0.1",
+        "alabaster",
+        "google-cloud-aiplatform[prediction]",
+        "recommonmark",
+        "gcp-sphinx-docfx-yaml",
     )
 
     shutil.rmtree(os.path.join("docs", "_build"), ignore_errors=True)
diff --git a/owlbot.py b/owlbot.py
index 45c5dda553..5e01104c8f 100644
--- a/owlbot.py
+++ b/owlbot.py
@@ -142,4 +142,11 @@
 # Don't include tests in calculation of test coverage
 s.replace("noxfile.py", """        \"--cov=tests/unit\",""", "")
 
+# Include prediction to be installed for documentation.
+s.replace(
+    "noxfile.py",
+    "\"alabaster\"",
+    "\"alabaster\", \"google-cloud-aiplatform[prediction]\"",
+)
+
 s.shell.run(["nox", "-s", "blacken"], hide_output=False)