From bf5104fcff6a6c2db8d8e39522c04eca1bb3fc93 Mon Sep 17 00:00:00 2001 From: Googler Date: Tue, 18 Jun 2024 11:50:26 -0700 Subject: [PATCH] feat(components): Update Starry Net image tags Signed-off-by: Googler PiperOrigin-RevId: 644464195 --- .../starry_net/get_training_artifacts/component.py | 2 +- .../_implementation/starry_net/version.py | 6 +++--- .../preview/starry_net/component.py | 9 ++++----- 3 files changed, 8 insertions(+), 9 deletions(-) diff --git a/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/get_training_artifacts/component.py b/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/get_training_artifacts/component.py index f1dc01b4008..da9ba8fc6b0 100644 --- a/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/get_training_artifacts/component.py +++ b/components/google-cloud/google_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/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/version.py b/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/version.py index ebb5b7be2de..0bf06f8790d 100644 --- a/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/version.py +++ b/components/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 index b50f6bf3076..039dd0fc18b 100644 --- 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