feat: adds enhanced protos for time series forecasting (#356)

Library generation was broken for a while and some commits were not auto proposed as PRs. This PR manually generated those changes. fix!: remove display_name from FeatureStore feat: add invalid_row_count to ImportFeatureValuesResponse and ImportFeatureValuesOperationMetadata
googleapis · May 17, 2021 · d73ec5a · d73ec5a
1 parent 0ab8e6b
commit d73ec5a
Show file tree

Hide file tree

Showing 36 changed files with 16,292 additions and 2,606 deletions.
diff --git a/...platform/src/main/java/com/google/cloud/aiplatform/v1beta1/FeaturestoreServiceClient.java b/...platform/src/main/java/com/google/cloud/aiplatform/v1beta1/FeaturestoreServiceClient.java
@@ -580,7 +580,7 @@ public final ListFeaturestoresPagedResponse listFeaturestores(ListFeaturestoresR
    *     be overwritten. Set the update_mask to `&#42;` to override all fields.
    *     <p>Updatable fields:
    *     <p>&#42; `display_name` &#42; `labels` &#42; `online_serving_config.fixed_node_count` &#42;
-   *     `online_serving_config.max_online_serving_size`
+   *     `retention_policy.online_storage_ttl_days`
    * @throws com.google.api.gax.rpc.ApiException if the remote call fails
    */
   public final OperationFuture<Featurestore, UpdateFeaturestoreOperationMetadata>

diff --git a/...form/src/test/java/com/google/cloud/aiplatform/v1beta1/FeaturestoreServiceClientTest.java b/...form/src/test/java/com/google/cloud/aiplatform/v1beta1/FeaturestoreServiceClientTest.java
@@ -95,7 +95,6 @@ public void createFeaturestoreTest() throws Exception {
     Featurestore expectedResponse =
         Featurestore.newBuilder()
             .setName(FeaturestoreName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]").toString())
-            .setDisplayName("displayName1714148973")
             .setCreateTime(Timestamp.newBuilder().build())
             .setUpdateTime(Timestamp.newBuilder().build())
             .setEtag("etag3123477")
@@ -150,7 +149,6 @@ public void createFeaturestoreTest2() throws Exception {
     Featurestore expectedResponse =
         Featurestore.newBuilder()
             .setName(FeaturestoreName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]").toString())
-            .setDisplayName("displayName1714148973")
             .setCreateTime(Timestamp.newBuilder().build())
             .setUpdateTime(Timestamp.newBuilder().build())
             .setEtag("etag3123477")
@@ -205,7 +203,6 @@ public void getFeaturestoreTest() throws Exception {
     Featurestore expectedResponse =
         Featurestore.newBuilder()
             .setName(FeaturestoreName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]").toString())
-            .setDisplayName("displayName1714148973")
             .setCreateTime(Timestamp.newBuilder().build())
             .setUpdateTime(Timestamp.newBuilder().build())
             .setEtag("etag3123477")
@@ -249,7 +246,6 @@ public void getFeaturestoreTest2() throws Exception {
     Featurestore expectedResponse =
         Featurestore.newBuilder()
             .setName(FeaturestoreName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]").toString())
-            .setDisplayName("displayName1714148973")
             .setCreateTime(Timestamp.newBuilder().build())
             .setUpdateTime(Timestamp.newBuilder().build())
             .setEtag("etag3123477")
@@ -381,7 +377,6 @@ public void updateFeaturestoreTest() throws Exception {
     Featurestore expectedResponse =
         Featurestore.newBuilder()
             .setName(FeaturestoreName.of("[PROJECT]", "[LOCATION]", "[FEATURESTORE]").toString())
-            .setDisplayName("displayName1714148973")
             .setCreateTime(Timestamp.newBuilder().build())
             .setUpdateTime(Timestamp.newBuilder().build())
             .setEtag("etag3123477")
@@ -1495,6 +1490,7 @@ public void importFeatureValuesTest() throws Exception {
         ImportFeatureValuesResponse.newBuilder()
             .setImportedEntityCount(-1840044274)
             .setImportedFeatureValueCount(-1221471427)
+            .setInvalidRowCount(-366655966)
             .build();
     Operation resultOperation =
         Operation.newBuilder()
@@ -1544,6 +1540,7 @@ public void importFeatureValuesTest2() throws Exception {
         ImportFeatureValuesResponse.newBuilder()
             .setImportedEntityCount(-1840044274)
             .setImportedFeatureValueCount(-1221471427)
+            .setInvalidRowCount(-366655966)
             .build();
     Operation resultOperation =
         Operation.newBuilder()

diff --git a/proto-google-cloud-aiplatform-v1/clirr-ignored-differences.xml b/proto-google-cloud-aiplatform-v1/clirr-ignored-differences.xml
@@ -16,4 +16,5 @@
     <differenceType>7012</differenceType>
     <method>* has*()</method>
   </difference>
+
 </differences>
diff --git a/proto-google-cloud-aiplatform-v1beta1/clirr-ignored-differences.xml b/proto-google-cloud-aiplatform-v1beta1/clirr-ignored-differences.xml
@@ -135,4 +135,49 @@
     <differenceType>6001</differenceType>
     <field>TPU*</field>
   </difference>
+  <difference>
+    <differenceType>7002</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <method>* get*(*)</method>
+  </difference>
+  <difference>
+    <differenceType>7002</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <method>* set*(*)</method>
+  </difference>
+  <difference>
+    <differenceType>7002</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <method>* has*(*)</method>
+  </difference>
+  <difference>
+    <differenceType>7002</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <method>* clear*(*)</method>
+  </difference>
+  <difference>
+    <differenceType>7002</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <method>* merge*(*)</method>
+  </difference>
+  <difference>
+    <differenceType>7002</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <method>* add*(*)</method>
+  </difference>
+  <difference>
+    <differenceType>6001</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <field>REPEATED_*</field>
+  </difference>
+  <difference>
+    <differenceType>6011</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <field>REPEATED_*</field>
+  </difference>
+  <difference>
+    <differenceType>6011</differenceType>
+    <className>com/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/AutoMlForecastingInputs*</className>
+    <field>*_FIELD_NUMBER</field>
+  </difference>
 </differences>
diff --git a/...aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorType.java b/...aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorType.java
@@ -88,6 +88,16 @@ public enum AcceleratorType implements com.google.protobuf.ProtocolMessageEnum {
    * <code>NVIDIA_TESLA_T4 = 5;</code>
    */
   NVIDIA_TESLA_T4(5),
+  /**
+   *
+   *
+   * <pre>
+   * Nvidia Tesla A100 GPU.
+   * </pre>
+   *
+   * <code>NVIDIA_TESLA_A100 = 8;</code>
+   */
+  NVIDIA_TESLA_A100(8),
   UNRECOGNIZED(-1),
   ;
 
@@ -151,6 +161,16 @@ public enum AcceleratorType implements com.google.protobuf.ProtocolMessageEnum {
    * <code>NVIDIA_TESLA_T4 = 5;</code>
    */
   public static final int NVIDIA_TESLA_T4_VALUE = 5;
+  /**
+   *
+   *
+   * <pre>
+   * Nvidia Tesla A100 GPU.
+   * </pre>
+   *
+   * <code>NVIDIA_TESLA_A100 = 8;</code>
+   */
+  public static final int NVIDIA_TESLA_A100_VALUE = 8;
 
   public final int getNumber() {
     if (this == UNRECOGNIZED) {
@@ -188,6 +208,8 @@ public static AcceleratorType forNumber(int value) {
         return NVIDIA_TESLA_P4;
       case 5:
         return NVIDIA_TESLA_T4;
+      case 8:
+        return NVIDIA_TESLA_A100;
       default:
         return null;
     }

diff --git a/...tform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorTypeProto.java b/...tform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorTypeProto.java
@@ -38,14 +38,15 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
       "\n6google/cloud/aiplatform/v1beta1/accele"
           + "rator_type.proto\022\037google.cloud.aiplatfor"
           + "m.v1beta1\032\034google/api/annotations.proto*"
-          + "\241\001\n\017AcceleratorType\022 \n\034ACCELERATOR_TYPE_"
+          + "\270\001\n\017AcceleratorType\022 \n\034ACCELERATOR_TYPE_"
           + "UNSPECIFIED\020\000\022\024\n\020NVIDIA_TESLA_K80\020\001\022\025\n\021N"
           + "VIDIA_TESLA_P100\020\002\022\025\n\021NVIDIA_TESLA_V100\020"
           + "\003\022\023\n\017NVIDIA_TESLA_P4\020\004\022\023\n\017NVIDIA_TESLA_T"
-          + "4\020\005B\210\001\n#com.google.cloud.aiplatform.v1be"
-          + "ta1B\024AcceleratorTypeProtoP\001ZIgoogle.gola"
-          + "ng.org/genproto/googleapis/cloud/aiplatf"
-          + "orm/v1beta1;aiplatformb\006proto3"
+          + "4\020\005\022\025\n\021NVIDIA_TESLA_A100\020\010B\210\001\n#com.googl"
+          + "e.cloud.aiplatform.v1beta1B\024AcceleratorT"
+          + "ypeProtoP\001ZIgoogle.golang.org/genproto/g"
+          + "oogleapis/cloud/aiplatform/v1beta1;aipla"
+          + "tformb\006proto3"
     };
     descriptor =
         com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(