feat(components): Update Starry Net image tags

Signed-off-by: Googler <nobody@google.com> PiperOrigin-RevId: 644464195
kubeflow · Jun 18, 2024 · bf5104f · bf5104f
1 parent 4dab191
commit bf5104f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 9 deletions.
diff --git a/..._cloud_pipeline_components/_implementation/starry_net/get_training_artifacts/component.py b/..._cloud_pipeline_components/_implementation/starry_net/get_training_artifacts/component.py
@@ -55,7 +55,7 @@ def get_training_artifacts(
       instance_schema_uri=str,
   )
   return outputs(
-      f'{docker_region}-docker.pkg.dev/vertex-ai/starryn/predictor:20240610_0542_RC00',  # pylint: disable=too-many-function-args
+      f'{docker_region}-docker.pkg.dev/vertex-ai/starryn/predictor:20240617_2142_RC00',  # pylint: disable=too-many-function-args
       private_dir,  # pylint: disable=too-many-function-args
       os.path.join(private_dir, 'predict_schema.yaml'),  # pylint: disable=too-many-function-args
       os.path.join(private_dir, 'instance_schema.yaml'),  # pylint: disable=too-many-function-args

diff --git a/...nents/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/version.py b/...nents/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/version.py
@@ -13,6 +13,6 @@
 # limitations under the License.
 """Version constants for starry net components."""
 
-DATAPREP_VERSION = '20240609_1425_RC00'
-PREDICTOR_VERSION = '20240610_0542_RC00'
-TRAINER_VERSION = '20240610_0542_RC00'
+DATAPREP_VERSION = '20240617_2225_RC00'
+PREDICTOR_VERSION = '20240617_2142_RC00'
+TRAINER_VERSION = '20240617_2142_RC00'
diff --git a/components/google-cloud/google_cloud_pipeline_components/preview/starry_net/component.py b/components/google-cloud/google_cloud_pipeline_components/preview/starry_net/component.py
@@ -15,7 +15,7 @@
 
 from typing import List
 
-#pylint: disable=g-importing-member
+# pylint: disable=g-importing-member
 from google_cloud_pipeline_components import _placeholders
 from google_cloud_pipeline_components._implementation.starry_net import DataprepOp
 from google_cloud_pipeline_components._implementation.starry_net import EvaluationOp
@@ -136,10 +136,9 @@ def starry_net(  # pylint: disable=dangerous-default-value
        dataprep_test_set_bigquery_dataset.
     dataprep_ts_identifier_columns: The list of ts_identifier columns from the
       BigQuery data source. These columns are used to distinguish the different
-      time series, so that if multiple rows have identical ts_identifier columns,
-      the series is generated by summing the target columns  for each timestamp.
-      If the big query dataset has a different  This is only used if
-      dataprep_bigquery_data_path is set.
+      time series, so that if multiple rows have identical ts_identifier
+      columns, the series is generated by summing the target columns for each
+      timestamp. This is only used if dataprep_bigquery_data_path is set.
     dataprep_time_column: The time column from the BigQuery data source. This is
       only used if dataprep_bigquery_data_path is set.
     dataprep_target_column: The column to be forecasted from the BigQuery data