feat: moves manual enhanced lib edits outside of generated files (#198)

telpirion · web-flow · commit a04a5613cec3 · 2021-02-08T11:01:59.000-08:00
diff --git a/google/cloud/aiplatform/gapic/schema/__init__.py b/google/cloud/aiplatform/gapic/schema/__init__.py
@@ -14,12 +14,33 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
+from google.cloud.aiplatform.helpers import _decorators
 from google.cloud.aiplatform.v1beta1.schema import predict
 from google.cloud.aiplatform.v1beta1.schema import trainingjob
-
+from google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1 import (
+    types as instance,
+)
+from google.cloud.aiplatform.v1beta1.schema.predict.params_v1beta1 import (
+    types as params,
+)
+from google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1 import (
+    types as prediction,
+)
+from google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1 import (
+    types as definition,
+)
 
 __all__ = (
     "predict",
     "trainingjob",
 )
+
+enhanced_types_packages = [
+    instance,
+    params,
+    prediction,
+    definition,
+]
+
+for pkg in enhanced_types_packages:
+    _decorators._add_methods_to_classes_in_package(pkg)
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/instance/__init__.py b/google/cloud/aiplatform/v1beta1/schema/predict/instance/__init__.py
@@ -14,9 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from google.cloud.aiplatform.helpers import _decorators
-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,
 )
@@ -56,4 +53,3 @@
     "VideoClassificationPredictionInstance",
     "VideoObjectTrackingPredictionInstance",
 )
-_decorators._add_methods_to_classes_in_package(pkg)
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/params/__init__.py b/google/cloud/aiplatform/v1beta1/schema/predict/params/__init__.py
@@ -14,9 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from google.cloud.aiplatform.helpers import _decorators
-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,
 )
@@ -44,4 +41,3 @@
     "VideoClassificationPredictionParams",
     "VideoObjectTrackingPredictionParams",
 )
-_decorators._add_methods_to_classes_in_package(pkg)
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/prediction/__init__.py b/google/cloud/aiplatform/v1beta1/schema/predict/prediction/__init__.py
@@ -14,11 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from google.cloud.aiplatform.helpers import _decorators
-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,
 )
@@ -66,4 +61,3 @@
     "VideoClassificationPredictionResult",
     "VideoObjectTrackingPredictionResult",
 )
-_decorators._add_methods_to_classes_in_package(pkg)
diff --git a/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/__init__.py b/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/__init__.py
@@ -14,11 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from google.cloud.aiplatform.helpers import _decorators
-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,
 )
@@ -134,4 +129,3 @@
     "AutoMlVideoObjectTrackingInputs",
     "ExportEvaluatedDataItemsConfig",
 )
-_decorators._add_methods_to_classes_in_package(pkg)

Original file line number	Diff line number	Diff line change
`@@ -14,9 +14,6 @@`
`14`	`14`	`# See the License for the specific language governing permissions and`
`15`	`15`	`# limitations under the License.`
`16`	`16`	`#`
`17`		`-from google.cloud.aiplatform.helpers import _decorators`
`18`		`-from google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1 import types as pkg`
`19`		`-`
`20`	`17`	`from google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1.types.image_classification import (`
`21`	`18`	`ImageClassificationPredictionInstance,`
`22`	`19`	`)`
`@@ -56,4 +53,3 @@`
`56`	`53`	`"VideoClassificationPredictionInstance",`
`57`	`54`	`"VideoObjectTrackingPredictionInstance",`
`58`	`55`	`)`
`59`		`-_decorators._add_methods_to_classes_in_package(pkg)`
Original file line number	Diff line number	Diff line change
`@@ -14,11 +14,6 @@`
`14`	`14`	`# See the License for the specific language governing permissions and`
`15`	`15`	`# limitations under the License.`
`16`	`16`	`#`
`17`		`-from google.cloud.aiplatform.helpers import _decorators`
`18`		`-from google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1 import (`
`19`		`- types as pkg,`
`20`		`-)`
`21`		`-`
`22`	`17`	`from google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1.types.classification import (`
`23`	`18`	`ClassificationPredictionResult,`
`24`	`19`	`)`
`@@ -66,4 +61,3 @@`
`66`	`61`	`"VideoClassificationPredictionResult",`
`67`	`62`	`"VideoObjectTrackingPredictionResult",`
`68`	`63`	`)`
`69`		`-_decorators._add_methods_to_classes_in_package(pkg)`