Merge pull request #305 from googleapis/dev

chore: merge dev into main for MB SDK Preview

Author: sasha-gitg

.coveragerc

@@ -2,10 +2,10 @@
 branch = True
 
 [report]
-fail_under = 100
+fail_under = 99
 show_missing = True
 omit =
-    google/cloud/aiplatform/v1/schema/trainingjob/definition/__init__.py
+    .nox/*
 exclude_lines =
     # Re-enable the standard pragma
     pragma: NO COVER
@@ -15,4 +15,4 @@ exclude_lines =
     # This is added at the module level as a safeguard for if someone
     # generates the code and tries to run it without pip installing. This
     # makes it virtually impossible to test properly.
-    except pkg_resources.DistributionNotFound
+    except pkg_resources.DistributionNotFound

.gitignore

@@ -26,6 +26,7 @@ pip-log.txt
 
 # Unit test / coverage reports
 .coverage
+.coverage.*
 .nox
 .cache
 .pytest_cache

.kokoro/samples/python3.8/common.cfg

@@ -19,6 +19,12 @@ env_vars: {
     value: "py-3.8"
 }
 
+# Run tests located under tests/system
+env_vars: {
+    key: "RUN_SYSTEM_TESTS"
+    value: "true"
+}
+
 env_vars: {
     key: "TRAMPOLINE_BUILD_FILE"
     value: "github/python-aiplatform/.kokoro/test-samples.sh"

README.rst

@@ -1,6 +1,14 @@
 Python Client for Cloud AI Platform
 =================================================
 
+**Experimental**
+
+This is an Experimental release. Experiments are focused on validating a prototype. They are not guaranteed to be released and might be subject to backward-incompatible changes. They are not intended for production use or covered by any SLA,  support obligation, or deprecation policy. They are covered by the `Pre-GA Offerings Terms`_ of the Google Cloud Platform Terms of Services.
+
+.. _Pre-GA Offerings Terms:  https://cloud.google.com/terms/service-terms#1
+
+----
+
 |beta| |pypi| |versions|
 
 

docs/aiplatform.rst

@@ -0,0 +1,6 @@
+Google Cloud Aiplatform SDK
+=============================================
+
+.. automodule:: google.cloud.aiplatform
+    :members:
+    :show-inheritance:

docs/index.rst

@@ -7,6 +7,7 @@ API Reference
 .. toctree::
     :maxdepth: 2
 
+    aiplatform
     aiplatform_v1/services
     aiplatform_v1/types
 

google/cloud/aiplatform/__init__.py

@@ -16,6 +16,52 @@
 #
 
 from google.cloud.aiplatform import gapic
+from google.cloud.aiplatform import explain
 
+from google.cloud.aiplatform import initializer
+from google.cloud.aiplatform.datasets import (
+    ImageDataset,
+    TabularDataset,
+    TextDataset,
+    VideoDataset,
+)
+from google.cloud.aiplatform.models import Endpoint
+from google.cloud.aiplatform.models import Model
+from google.cloud.aiplatform.jobs import BatchPredictionJob
+from google.cloud.aiplatform.training_jobs import (
+    CustomTrainingJob,
+    CustomContainerTrainingJob,
+    CustomPythonPackageTrainingJob,
+    AutoMLTabularTrainingJob,
+    AutoMLImageTrainingJob,
+    AutoMLTextTrainingJob,
+    AutoMLVideoTrainingJob,
+)
 
-__all__ = ("gapic",)
+"""
+Usage:
+from google.cloud import aiplatform
+
+aiplatform.init(project='my_project')
+"""
+init = initializer.global_config.init
+
+__all__ = (
+    "explain",
+    "gapic",
+    "init",
+    "AutoMLImageTrainingJob",
+    "AutoMLTabularTrainingJob",
+    "AutoMLTextTrainingJob",
+    "AutoMLVideoTrainingJob",
+    "BatchPredictionJob",
+    "CustomTrainingJob",
+    "CustomContainerTrainingJob",
+    "CustomPythonPackageTrainingJob",
+    "Endpoint",
+    "ImageDataset",
+    "Model",
+    "TabularDataset",
+    "TextDataset",
+    "VideoDataset",
+)