feat: add schema namespace (#140)

* feat: add schema namespace

Author: dizcology

google/cloud/aiplatform/__init__.py

@@ -16,5 +16,7 @@
 #
 
 from google.cloud.aiplatform import gapic
+from google.cloud.aiplatform import schema
 
-__all__ = (gapic,)
+
+__all__ = (gapic, schema)

google/cloud/aiplatform/schema/__init__.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from google.cloud.aiplatform.v1beta1.schema import predict
+from google.cloud.aiplatform.v1beta1.schema import trainingjob
+
+
+__all__ = (
+ "predict",
+ "trainingjob",
+)

google/cloud/aiplatform/v1beta1/schema/__init__.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from google.cloud.aiplatform.v1beta1.schema import predict
+from google.cloud.aiplatform.v1beta1.schema import trainingjob
+
+
+__all__ = (
+ "predict",
+ "trainingjob",
+)

google/cloud/aiplatform/v1beta1/schema/predict/__init__.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from google.cloud.aiplatform.v1beta1.schema.predict import instance
+from google.cloud.aiplatform.v1beta1.schema.predict import params
+from google.cloud.aiplatform.v1beta1.schema.predict import prediction
+
+__all__ = (
+ "instance",
+ "params",
+ "prediction",
+)

google/cloud/aiplatform/v1beta1/schema/predict/instance/__init__.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 #
 from google.cloud.aiplatform.helpers import _decorators
-import google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1.types as pkg
+from google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1 import types as pkg
 
 from google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1.types.image_classification import (
  ImageClassificationPredictionInstance,

google/cloud/aiplatform/v1beta1/schema/predict/params/__init__.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 #
 from google.cloud.aiplatform.helpers import _decorators
-import google.cloud.aiplatform.v1beta1.schema.predict.params_v1beta1.types as pkg
+from google.cloud.aiplatform.v1beta1.schema.predict.params_v1beta1 import types as pkg
 
 from google.cloud.aiplatform.v1beta1.schema.predict.params_v1beta1.types.image_classification import (
  ImageClassificationPredictionParams,

google/cloud/aiplatform/v1beta1/schema/predict/prediction/__init__.py

@@ -15,7 +15,9 @@
 # limitations under the License.
 #
 from google.cloud.aiplatform.helpers import _decorators
-import google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1.types as pkg
+from google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1 import (
+ types as pkg,
+)
 
 from google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1.types.classification import (
  ClassificationPredictionResult,

google/cloud/aiplatform/v1beta1/schema/trainingjob/__init__.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from google.cloud.aiplatform.v1beta1.schema.trainingjob import definition
+
+__all__ = ("definition",)

google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/__init__.py

@@ -15,7 +15,9 @@
 # limitations under the License.
 #
 from google.cloud.aiplatform.helpers import _decorators
-import google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types as pkg
+from google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1 import (
+ types as pkg,
+)
 
 from google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types.automl_forecasting import (
  AutoMlForecasting,

samples/snippets/create_training_pipeline_image_classification_sample.py

@@ -14,8 +14,7 @@
 
 # [START aiplatform_create_training_pipeline_image_classification_sample]
 from google.cloud import aiplatform
-from google.cloud.aiplatform.v1beta1.schema.trainingjob import definition
-ModelType = definition.AutoMlImageClassificationInputs().ModelType
+from google.cloud.aiplatform.schema import trainingjob
 
 
 def create_training_pipeline_image_classification_sample(
@@ -32,9 +31,9 @@ def create_training_pipeline_image_classification_sample(
  # This client only needs to be created once, and can be reused for multiple requests.
  client = aiplatform.gapic.PipelineServiceClient(client_options=client_options)
 
- icn_training_inputs = definition.AutoMlImageClassificationInputs(
+ icn_training_inputs = trainingjob.definition.AutoMlImageClassificationInputs(
  multi_label=True,
- model_type=ModelType.CLOUD,
+ model_type=trainingjob.definition.AutoMlImageClassificationInputs.ModelType.CLOUD,
  budget_milli_node_hours=8000,
  disable_early_stopping=False
  )