google-cloud-pipeline-components 2.20.1__py3-none-any.whl → 2.21.0__py3-none-any.whl

google_cloud_pipeline_components/preview/automl/tabular/tabnet_trainer_pipeline.yaml

@@ -2875,7 +2875,7 @@ deploymentSpec:
           \"encryption_spec\": {\"kms_key_name\":\"", "{{$.inputs.parameters[''encryption_spec_key_name'']}}",
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\": 1, \"machine_spec\":
           {\"machine_type\": \"n1-standard-8\"}, \"container_spec\": {\"image_uri\":\"",
-          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250129_0625", "\",
+          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250827_0525", "\",
           \"args\": [\"cancel_l2l_tuner\", \"--error_file_path=", "{{$.inputs.parameters[''root_dir'']}}",
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/error.pb\", \"--cleanup_lro_job_infos=",
           "{{$.inputs.parameters[''root_dir'']}}", "/{{$.pipeline_job_uuid}}/lro\"]}}]}}"]}'
@@ -2890,7 +2890,7 @@ deploymentSpec:
         args:
         - --executor_input
         - '{{$}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 52.0
@@ -2915,7 +2915,7 @@ deploymentSpec:
           \ *\n\ndef _bool_identity(value: bool) -> str:\n  \"\"\"Returns boolean\
           \ value.\n\n  Args:\n    value: Boolean value to return\n\n  Returns:\n\
           \    Boolean value.\n  \"\"\"\n  return 'true' if value else 'false'\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-feature-transform-engine:
       container:
         args:
@@ -3000,8 +3000,8 @@ deploymentSpec:
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/dataflow_tmp"]}'
         - '{"Concat": ["--dataflow_max_num_workers=", "{{$.inputs.parameters[''dataflow_max_num_workers'']}}"]}'
         - '{"Concat": ["--dataflow_machine_type=", "{{$.inputs.parameters[''dataflow_machine_type'']}}"]}'
-        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
-        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
+        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         - '{"Concat": ["--dataflow_disk_size_gb=", "{{$.inputs.parameters[''dataflow_disk_size_gb'']}}"]}'
         - '{"Concat": ["--dataflow_subnetwork_fully_qualified=", "{{$.inputs.parameters[''dataflow_subnetwork'']}}"]}'
         - '{"Concat": ["--dataflow_use_public_ips=", "{{$.inputs.parameters[''dataflow_use_public_ips'']}}"]}'
@@ -3018,7 +3018,7 @@ deploymentSpec:
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
         - '{"Concat": ["--encryption_spec_key_name=", "{{$.inputs.parameters[''encryption_spec_key_name'']}}"]}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 30.0
@@ -3048,7 +3048,7 @@ deploymentSpec:
           \n  return collections.namedtuple(\n      'Outputs',\n      [\n        \
           \  'model_display_name',\n      ],\n  )(\n      model_display_name,\n  )\n\
           \n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-model-batch-predict:
       container:
         args:
@@ -3289,7 +3289,7 @@ deploymentSpec:
           \   'training_disk_spec',\n          'eval_machine_spec',\n          'eval_replica_count',\n\
           \      ],\n  )(\n      training_machine_spec,\n      training_disk_spec,\n\
           \      eval_machine_spec,\n      eval_replica_count,\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-set-optional-inputs:
       container:
         args:
@@ -3337,7 +3337,7 @@ deploymentSpec:
           \ 'data_source_csv_filenames',\n          'data_source_bigquery_table_path',\n\
           \      ],\n  )(\n      data_source_csv_filenames,\n      data_source_bigquery_table_path,\n\
           \  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-split-materialized-data:
       container:
         args:
@@ -3383,7 +3383,7 @@ deploymentSpec:
           \ 'w') as f:\n    f.write(file_patterns[0])\n\n  with tf.io.gfile.GFile(materialized_eval_split,\
           \ 'w') as f:\n    f.write(file_patterns[1])\n\n  with tf.io.gfile.GFile(materialized_test_split,\
           \ 'w') as f:\n    f.write(file_patterns[2])\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
     exec-tabnet-trainer:
       container:
         args:
@@ -3401,11 +3401,11 @@ deploymentSpec:
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\":\"", "1",
           "\", \"machine_spec\": ", "{{$.inputs.parameters[''training_machine_spec'']}}",
           ", \"disk_spec\": ", "{{$.inputs.parameters[''training_disk_spec'']}}",
-          ", \"container_spec\": {\"image_uri\":\"", "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/tabnet-training:20250129_0625",
+          ", \"container_spec\": {\"image_uri\":\"", "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/tabnet-training:20250827_0525",
           "\", \"args\": [\"--target_column=", "{{$.inputs.parameters[''target_column'']}}",
           "\", \"--weight_column=", "{{$.inputs.parameters[''weight_column'']}}",
           "\", \"--model_type=", "{{$.inputs.parameters[''prediction_type'']}}", "\",
-          \"--prediction_docker_uri=", "us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625",
+          \"--prediction_docker_uri=", "us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525",
           "\", \"--baseline_path=", "{{$.inputs.artifacts[''instance_baseline''].uri}}",
           "\", \"--metadata_path=", "{{$.inputs.artifacts[''metadata''].uri}}", "\",
           \"--transform_output_path=", "{{$.inputs.artifacts[''transform_output''].uri}}",
@@ -3492,7 +3492,7 @@ deploymentSpec:
           ["--temporal_total_weight=", "{{$.inputs.parameters[''temporal_total_weight'']}}"]}}}'
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
 pipelineInfo:
   description: 'Train a model using the Tabular Workflow for TabNet pipelines.
 

google_cloud_pipeline_components/preview/automl/tabular/wide_and_deep_hyperparameter_tuning_job.py

@@ -158,7 +158,7 @@ def wide_and_deep_hyperparameter_tuning_job(
                   ', "disk_spec": ',
                   training_disk_spec,
                   ', "container_spec": {"image_uri":"',
-                  'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250129_0625',
+                  'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250827_0525',
                   '", "args": ["--target_column=',
                   target_column,
                   '", "--weight_column=',
@@ -166,7 +166,7 @@ def wide_and_deep_hyperparameter_tuning_job(
                   '", "--model_type=',
                   prediction_type,
                   '", "--prediction_docker_uri=',
-                  'us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625',
+                  'us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525',
                   '", "--prediction_docker_uri_artifact_path=',
                   prediction_docker_uri_output,
                   '", "--baseline_path=',

google_cloud_pipeline_components/preview/automl/tabular/wide_and_deep_hyperparameter_tuning_job_pipeline.yaml

@@ -2632,7 +2632,7 @@ deploymentSpec:
           \"encryption_spec\": {\"kms_key_name\":\"", "{{$.inputs.parameters[''encryption_spec_key_name'']}}",
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\": 1, \"machine_spec\":
           {\"machine_type\": \"n1-standard-8\"}, \"container_spec\": {\"image_uri\":\"",
-          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250129_0625", "\",
+          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250827_0525", "\",
           \"args\": [\"cancel_l2l_tuner\", \"--error_file_path=", "{{$.inputs.parameters[''root_dir'']}}",
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/error.pb\", \"--cleanup_lro_job_infos=",
           "{{$.inputs.parameters[''root_dir'']}}", "/{{$.pipeline_job_uuid}}/lro\"]}}]}}"]}'
@@ -2647,7 +2647,7 @@ deploymentSpec:
         args:
         - --executor_input
         - '{{$}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 52.0
@@ -2672,7 +2672,7 @@ deploymentSpec:
           \ *\n\ndef _bool_identity(value: bool) -> str:\n  \"\"\"Returns boolean\
           \ value.\n\n  Args:\n    value: Boolean value to return\n\n  Returns:\n\
           \    Boolean value.\n  \"\"\"\n  return 'true' if value else 'false'\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-feature-transform-engine:
       container:
         args:
@@ -2757,8 +2757,8 @@ deploymentSpec:
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/dataflow_tmp"]}'
         - '{"Concat": ["--dataflow_max_num_workers=", "{{$.inputs.parameters[''dataflow_max_num_workers'']}}"]}'
         - '{"Concat": ["--dataflow_machine_type=", "{{$.inputs.parameters[''dataflow_machine_type'']}}"]}'
-        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
-        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
+        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         - '{"Concat": ["--dataflow_disk_size_gb=", "{{$.inputs.parameters[''dataflow_disk_size_gb'']}}"]}'
         - '{"Concat": ["--dataflow_subnetwork_fully_qualified=", "{{$.inputs.parameters[''dataflow_subnetwork'']}}"]}'
         - '{"Concat": ["--dataflow_use_public_ips=", "{{$.inputs.parameters[''dataflow_use_public_ips'']}}"]}'
@@ -2775,7 +2775,7 @@ deploymentSpec:
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
         - '{"Concat": ["--encryption_spec_key_name=", "{{$.inputs.parameters[''encryption_spec_key_name'']}}"]}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 30.0
@@ -2843,7 +2843,7 @@ deploymentSpec:
           \ = {\n      'instanceSchemaUri': instance_schema_uri,\n      'predictionSchemaUri':\
           \ prediction_schema_uri,\n  }\n  unmanaged_container_model.uri = os.path.join(\n\
           \      trials_dir, 'trial_{}'.format(best_trial['id']), 'model'\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-model-display-name:
       container:
         args:
@@ -2870,7 +2870,7 @@ deploymentSpec:
           \n  return collections.namedtuple(\n      'Outputs',\n      [\n        \
           \  'model_display_name',\n      ],\n  )(\n      model_display_name,\n  )\n\
           \n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-wide-and-deep-study-spec-parameters:
       container:
         args:
@@ -3147,7 +3147,7 @@ deploymentSpec:
           \   'training_disk_spec',\n          'eval_machine_spec',\n          'eval_replica_count',\n\
           \      ],\n  )(\n      training_machine_spec,\n      training_disk_spec,\n\
           \      eval_machine_spec,\n      eval_replica_count,\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-set-optional-inputs:
       container:
         args:
@@ -3195,7 +3195,7 @@ deploymentSpec:
           \ 'data_source_csv_filenames',\n          'data_source_bigquery_table_path',\n\
           \      ],\n  )(\n      data_source_csv_filenames,\n      data_source_bigquery_table_path,\n\
           \  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-split-materialized-data:
       container:
         args:
@@ -3241,7 +3241,7 @@ deploymentSpec:
           \ 'w') as f:\n    f.write(file_patterns[0])\n\n  with tf.io.gfile.GFile(materialized_eval_split,\
           \ 'w') as f:\n    f.write(file_patterns[1])\n\n  with tf.io.gfile.GFile(materialized_test_split,\
           \ 'w') as f:\n    f.write(file_patterns[2])\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
     exec-training-configurator-and-validator:
       container:
         args:
@@ -3286,7 +3286,7 @@ deploymentSpec:
           ["--temporal_total_weight=", "{{$.inputs.parameters[''temporal_total_weight'']}}"]}}}'
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
     exec-wide-and-deep-hyperparameter-tuning-job:
       container:
         args:
@@ -3314,11 +3314,11 @@ deploymentSpec:
           ", \"trial_job_spec\": {\"worker_pool_specs\": [{\"replica_count\":\"",
           "1", "\", \"machine_spec\": ", "{{$.inputs.parameters[''training_machine_spec'']}}",
           ", \"disk_spec\": ", "{{$.inputs.parameters[''training_disk_spec'']}}",
-          ", \"container_spec\": {\"image_uri\":\"", "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250129_0625",
+          ", \"container_spec\": {\"image_uri\":\"", "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250827_0525",
           "\", \"args\": [\"--target_column=", "{{$.inputs.parameters[''target_column'']}}",
           "\", \"--weight_column=", "{{$.inputs.parameters[''weight_column'']}}",
           "\", \"--model_type=", "{{$.inputs.parameters[''prediction_type'']}}", "\",
-          \"--prediction_docker_uri=", "us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625",
+          \"--prediction_docker_uri=", "us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525",
           "\", \"--prediction_docker_uri_artifact_path=", "{{$.outputs.parameters[''prediction_docker_uri_output''].output_file}}",
           "\", \"--baseline_path=", "{{$.inputs.artifacts[''instance_baseline''].uri}}",
           "\", \"--metadata_path=", "{{$.inputs.artifacts[''metadata''].uri}}", "\",

google_cloud_pipeline_components/preview/automl/tabular/wide_and_deep_trainer.py

@@ -161,7 +161,7 @@ def wide_and_deep_trainer(
                   ', "disk_spec": ',
                   training_disk_spec,
                   ', "container_spec": {"image_uri":"',
-                  'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250129_0625',
+                  'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250827_0525',
                   '", "args": ["--target_column=',
                   target_column,
                   '", "--weight_column=',
@@ -169,7 +169,7 @@ def wide_and_deep_trainer(
                   '", "--model_type=',
                   prediction_type,
                   '", "--prediction_docker_uri=',
-                  'us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625',
+                  'us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525',
                   '", "--baseline_path=',
                   instance_baseline.uri,
                   '", "--metadata_path=',

google_cloud_pipeline_components/preview/automl/tabular/wide_and_deep_trainer_pipeline.yaml

@@ -2674,7 +2674,7 @@ deploymentSpec:
           \"encryption_spec\": {\"kms_key_name\":\"", "{{$.inputs.parameters[''encryption_spec_key_name'']}}",
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\": 1, \"machine_spec\":
           {\"machine_type\": \"n1-standard-8\"}, \"container_spec\": {\"image_uri\":\"",
-          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250129_0625", "\",
+          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250827_0525", "\",
           \"args\": [\"cancel_l2l_tuner\", \"--error_file_path=", "{{$.inputs.parameters[''root_dir'']}}",
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/error.pb\", \"--cleanup_lro_job_infos=",
           "{{$.inputs.parameters[''root_dir'']}}", "/{{$.pipeline_job_uuid}}/lro\"]}}]}}"]}'
@@ -2689,7 +2689,7 @@ deploymentSpec:
         args:
         - --executor_input
         - '{{$}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 52.0
@@ -2714,7 +2714,7 @@ deploymentSpec:
           \ *\n\ndef _bool_identity(value: bool) -> str:\n  \"\"\"Returns boolean\
           \ value.\n\n  Args:\n    value: Boolean value to return\n\n  Returns:\n\
           \    Boolean value.\n  \"\"\"\n  return 'true' if value else 'false'\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-feature-transform-engine:
       container:
         args:
@@ -2799,8 +2799,8 @@ deploymentSpec:
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/dataflow_tmp"]}'
         - '{"Concat": ["--dataflow_max_num_workers=", "{{$.inputs.parameters[''dataflow_max_num_workers'']}}"]}'
         - '{"Concat": ["--dataflow_machine_type=", "{{$.inputs.parameters[''dataflow_machine_type'']}}"]}'
-        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
-        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
+        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         - '{"Concat": ["--dataflow_disk_size_gb=", "{{$.inputs.parameters[''dataflow_disk_size_gb'']}}"]}'
         - '{"Concat": ["--dataflow_subnetwork_fully_qualified=", "{{$.inputs.parameters[''dataflow_subnetwork'']}}"]}'
         - '{"Concat": ["--dataflow_use_public_ips=", "{{$.inputs.parameters[''dataflow_use_public_ips'']}}"]}'
@@ -2817,7 +2817,7 @@ deploymentSpec:
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
         - '{"Concat": ["--encryption_spec_key_name=", "{{$.inputs.parameters[''encryption_spec_key_name'']}}"]}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 30.0
@@ -2847,7 +2847,7 @@ deploymentSpec:
           \n  return collections.namedtuple(\n      'Outputs',\n      [\n        \
           \  'model_display_name',\n      ],\n  )(\n      model_display_name,\n  )\n\
           \n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-model-batch-predict:
       container:
         args:
@@ -3040,7 +3040,7 @@ deploymentSpec:
           \   'training_disk_spec',\n          'eval_machine_spec',\n          'eval_replica_count',\n\
           \      ],\n  )(\n      training_machine_spec,\n      training_disk_spec,\n\
           \      eval_machine_spec,\n      eval_replica_count,\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-set-optional-inputs:
       container:
         args:
@@ -3088,7 +3088,7 @@ deploymentSpec:
           \ 'data_source_csv_filenames',\n          'data_source_bigquery_table_path',\n\
           \      ],\n  )(\n      data_source_csv_filenames,\n      data_source_bigquery_table_path,\n\
           \  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-split-materialized-data:
       container:
         args:
@@ -3134,7 +3134,7 @@ deploymentSpec:
           \ 'w') as f:\n    f.write(file_patterns[0])\n\n  with tf.io.gfile.GFile(materialized_eval_split,\
           \ 'w') as f:\n    f.write(file_patterns[1])\n\n  with tf.io.gfile.GFile(materialized_test_split,\
           \ 'w') as f:\n    f.write(file_patterns[2])\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
     exec-training-configurator-and-validator:
       container:
         args:
@@ -3179,7 +3179,7 @@ deploymentSpec:
           ["--temporal_total_weight=", "{{$.inputs.parameters[''temporal_total_weight'']}}"]}}}'
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
     exec-wide-and-deep-trainer:
       container:
         args:
@@ -3197,11 +3197,11 @@ deploymentSpec:
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\":\"", "1",
           "\", \"machine_spec\": ", "{{$.inputs.parameters[''training_machine_spec'']}}",
           ", \"disk_spec\": ", "{{$.inputs.parameters[''training_disk_spec'']}}",
-          ", \"container_spec\": {\"image_uri\":\"", "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250129_0625",
+          ", \"container_spec\": {\"image_uri\":\"", "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/wide-and-deep-training:20250827_0525",
           "\", \"args\": [\"--target_column=", "{{$.inputs.parameters[''target_column'']}}",
           "\", \"--weight_column=", "{{$.inputs.parameters[''weight_column'']}}",
           "\", \"--model_type=", "{{$.inputs.parameters[''prediction_type'']}}", "\",
-          \"--prediction_docker_uri=", "us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250129_0625",
+          \"--prediction_docker_uri=", "us-docker.pkg.dev/vertex-ai/automl-tabular/prediction-server:20250827_0525",
           "\", \"--baseline_path=", "{{$.inputs.artifacts[''instance_baseline''].uri}}",
           "\", \"--metadata_path=", "{{$.inputs.artifacts[''metadata''].uri}}", "\",
           \"--transform_output_path=", "{{$.inputs.artifacts[''transform_output''].uri}}",

google_cloud_pipeline_components/preview/automl/tabular/xgboost_hyperparameter_tuning_job_pipeline.yaml

@@ -2620,7 +2620,7 @@ deploymentSpec:
           \"encryption_spec\": {\"kms_key_name\":\"", "{{$.inputs.parameters[''encryption_spec_key_name'']}}",
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\": 1, \"machine_spec\":
           {\"machine_type\": \"n1-standard-8\"}, \"container_spec\": {\"image_uri\":\"",
-          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250129_0625", "\",
+          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250827_0525", "\",
           \"args\": [\"cancel_l2l_tuner\", \"--error_file_path=", "{{$.inputs.parameters[''root_dir'']}}",
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/error.pb\", \"--cleanup_lro_job_infos=",
           "{{$.inputs.parameters[''root_dir'']}}", "/{{$.pipeline_job_uuid}}/lro\"]}}]}}"]}'
@@ -2651,7 +2651,7 @@ deploymentSpec:
           \ *\n\ndef _bool_identity(value: bool) -> str:\n  \"\"\"Returns boolean\
           \ value.\n\n  Args:\n    value: Boolean value to return\n\n  Returns:\n\
           \    Boolean value.\n  \"\"\"\n  return 'true' if value else 'false'\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-feature-transform-engine:
       container:
         args:
@@ -2736,8 +2736,8 @@ deploymentSpec:
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/dataflow_tmp"]}'
         - '{"Concat": ["--dataflow_max_num_workers=", "{{$.inputs.parameters[''dataflow_max_num_workers'']}}"]}'
         - '{"Concat": ["--dataflow_machine_type=", "{{$.inputs.parameters[''dataflow_machine_type'']}}"]}'
-        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
-        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
+        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         - '{"Concat": ["--dataflow_disk_size_gb=", "{{$.inputs.parameters[''dataflow_disk_size_gb'']}}"]}'
         - '{"Concat": ["--dataflow_subnetwork_fully_qualified=", "{{$.inputs.parameters[''dataflow_subnetwork'']}}"]}'
         - '{"Concat": ["--dataflow_use_public_ips=", "{{$.inputs.parameters[''dataflow_use_public_ips'']}}"]}'
@@ -2754,7 +2754,7 @@ deploymentSpec:
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
         - '{"Concat": ["--encryption_spec_key_name=", "{{$.inputs.parameters[''encryption_spec_key_name'']}}"]}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 30.0
@@ -2818,7 +2818,7 @@ deploymentSpec:
           \    return re.sub(r'^/gcs/', r'gs://', path)\n\n  master_worker_pool_spec\
           \ = {\n      'replica_count': 1,\n      'machine_spec': {\n          'machine_type':\
           \ machine_type,\n      },\n      'container_spec': {\n          'image_uri':\
-          \ 'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/xgboost-training:20250129_0625',\n\
+          \ 'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/xgboost-training:20250827_0525',\n\
           \          'args': [\n              f'--job_dir={get_gcs_path(job_dir)}',\n\
           \              f'--instance_schema_path={get_gcs_path(instance_schema_uri)}',\n\
           \              f'--prediction_schema_path={get_gcs_path(prediction_schema_uri)}',\n\
@@ -2831,7 +2831,7 @@ deploymentSpec:
           \              f'--baseline_path={get_gcs_path(instance_baseline)}',\n \
           \             f'--eval_metric={eval_metric}',\n              f'--disable_default_eval_metric={disable_default_eval_metric}',\n\
           \              f'--seed={seed}',\n              f'--seed_per_iteration={seed_per_iteration}',\n\
-          \              '--prediction_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/xgboost-prediction-server:20250129_0625',\n\
+          \              '--prediction_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/xgboost-prediction-server:20250827_0525',\n\
           \          ],\n      },\n  }\n\n  # Add optional arguments if set\n  if\
           \ weight_column:\n    master_worker_pool_spec['container_spec']['args'].append(\n\
           \        f'--weight_column={weight_column}'\n    )\n\n  # Add accelerator_type\
@@ -2850,7 +2850,7 @@ deploymentSpec:
           \      ],\n  )(\n      worker_pool_specs_lst,\n      get_gcs_path(instance_schema_uri),\n\
           \      get_gcs_path(prediction_schema_uri),\n      get_gcs_path(trials),\n\
           \      get_gcs_path(prediction_docker_uri_output),\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-best-hyperparameter-tuning-job-trial:
       container:
         args:
@@ -2915,7 +2915,7 @@ deploymentSpec:
           \ = {\n      'instanceSchemaUri': instance_schema_uri,\n      'predictionSchemaUri':\
           \ prediction_schema_uri,\n  }\n  unmanaged_container_model.uri = os.path.join(\n\
           \      trials_dir, 'trial_{}'.format(best_trial['id']), 'model'\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-model-display-name:
       container:
         args:
@@ -2942,7 +2942,7 @@ deploymentSpec:
           \n  return collections.namedtuple(\n      'Outputs',\n      [\n        \
           \  'model_display_name',\n      ],\n  )(\n      model_display_name,\n  )\n\
           \n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-prediction-type-for-xgboost:
       container:
         args:
@@ -2971,7 +2971,7 @@ deploymentSpec:
           \ Must be one of'\n        ' [reg:squarederror, reg:squaredlogerror, reg:logistic,\
           \ reg:gamma,'\n        ' reg:tweedie, reg:pseudohubererror, binary:logistic,'\n\
           \        ' multi:softprob].'\n    )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-xgboost-study-spec-parameters:
       container:
         args:
@@ -3546,7 +3546,7 @@ deploymentSpec:
           \ 'data_source_csv_filenames',\n          'data_source_bigquery_table_path',\n\
           \      ],\n  )(\n      data_source_csv_filenames,\n      data_source_bigquery_table_path,\n\
           \  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-split-materialized-data:
       container:
         args:
@@ -3592,7 +3592,7 @@ deploymentSpec:
           \ 'w') as f:\n    f.write(file_patterns[0])\n\n  with tf.io.gfile.GFile(materialized_eval_split,\
           \ 'w') as f:\n    f.write(file_patterns[1])\n\n  with tf.io.gfile.GFile(materialized_test_split,\
           \ 'w') as f:\n    f.write(file_patterns[2])\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
     exec-training-configurator-and-validator:
       container:
         args:
@@ -3637,7 +3637,7 @@ deploymentSpec:
           ["--temporal_total_weight=", "{{$.inputs.parameters[''temporal_total_weight'']}}"]}}}'
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
     exec-xgboost-hyperparameter-tuning-job:
       container:
         args:

google_cloud_pipeline_components/preview/automl/tabular/xgboost_trainer_pipeline.yaml

@@ -2844,7 +2844,7 @@ deploymentSpec:
           \"encryption_spec\": {\"kms_key_name\":\"", "{{$.inputs.parameters[''encryption_spec_key_name'']}}",
           "\"}, \"job_spec\": {\"worker_pool_specs\": [{\"replica_count\": 1, \"machine_spec\":
           {\"machine_type\": \"n1-standard-8\"}, \"container_spec\": {\"image_uri\":\"",
-          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250129_0625", "\",
+          "us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/training:20250827_0525", "\",
           \"args\": [\"cancel_l2l_tuner\", \"--error_file_path=", "{{$.inputs.parameters[''root_dir'']}}",
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/error.pb\", \"--cleanup_lro_job_infos=",
           "{{$.inputs.parameters[''root_dir'']}}", "/{{$.pipeline_job_uuid}}/lro\"]}}]}}"]}'
@@ -2875,7 +2875,7 @@ deploymentSpec:
           \ *\n\ndef _bool_identity(value: bool) -> str:\n  \"\"\"Returns boolean\
           \ value.\n\n  Args:\n    value: Boolean value to return\n\n  Returns:\n\
           \    Boolean value.\n  \"\"\"\n  return 'true' if value else 'false'\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-feature-transform-engine:
       container:
         args:
@@ -2960,8 +2960,8 @@ deploymentSpec:
           "/{{$.pipeline_job_uuid}}/{{$.pipeline_task_uuid}}/dataflow_tmp"]}'
         - '{"Concat": ["--dataflow_max_num_workers=", "{{$.inputs.parameters[''dataflow_max_num_workers'']}}"]}'
         - '{"Concat": ["--dataflow_machine_type=", "{{$.inputs.parameters[''dataflow_machine_type'']}}"]}'
-        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
-        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        - --dataflow_worker_container_image=us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
+        - --feature_transform_engine_docker_uri=us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         - '{"Concat": ["--dataflow_disk_size_gb=", "{{$.inputs.parameters[''dataflow_disk_size_gb'']}}"]}'
         - '{"Concat": ["--dataflow_subnetwork_fully_qualified=", "{{$.inputs.parameters[''dataflow_subnetwork'']}}"]}'
         - '{"Concat": ["--dataflow_use_public_ips=", "{{$.inputs.parameters[''dataflow_use_public_ips'']}}"]}'
@@ -2978,7 +2978,7 @@ deploymentSpec:
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
         - '{"Concat": ["--encryption_spec_key_name=", "{{$.inputs.parameters[''encryption_spec_key_name'']}}"]}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
         resources:
           cpuLimit: 8.0
           memoryLimit: 30.0
@@ -3098,10 +3098,10 @@ deploymentSpec:
           \ worker pool specs.\n  \"\"\"\n  import copy\n  import collections\n  import\
           \ os\n  import re\n\n  def get_gcs_path(path):\n    return re.sub(r'/gcs/',\
           \ 'gs://', path)\n\n  formatted_job_dir = get_gcs_path(job_dir)\n  prediction_docker_uri\
-          \ = (\n      'us-docker.pkg.dev/vertex-ai/automl-tabular/xgboost-prediction-server:20250129_0625'\n\
+          \ = (\n      'us-docker.pkg.dev/vertex-ai/automl-tabular/xgboost-prediction-server:20250827_0525'\n\
           \  )\n  master_worker_pool_spec = {\n      'replica_count': 1,\n      'machine_spec':\
           \ {\n          'machine_type': machine_type,\n      },\n      'container_spec':\
-          \ {\n          'image_uri': 'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/xgboost-training:20250129_0625',\n\
+          \ {\n          'image_uri': 'us-docker.pkg.dev/vertex-ai-restricted/automl-tabular/xgboost-training:20250827_0525',\n\
           \          'args': [\n              f'--job_dir={formatted_job_dir}',\n\
           \              f'--target_column={target_column}',\n              f'--objective={objective}',\n\
           \              f'--training_data_path={get_gcs_path(materialized_train_split)}',\n\
@@ -3159,7 +3159,7 @@ deploymentSpec:
           \      'predictionSchemaUri': os.path.join(model_dir, 'prediction_schema.yaml'),\n\
           \  }\n  unmanaged_container_model.uri = model_dir\n\n  return collections.namedtuple('Outputs',\
           \ ['worker_pool_specs'])(\n      worker_pool_specs_lst\n  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-model-display-name:
       container:
         args:
@@ -3186,7 +3186,7 @@ deploymentSpec:
           \n  return collections.namedtuple(\n      'Outputs',\n      [\n        \
           \  'model_display_name',\n      ],\n  )(\n      model_display_name,\n  )\n\
           \n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-get-prediction-type-for-xgboost:
       container:
         args:
@@ -3215,7 +3215,7 @@ deploymentSpec:
           \ Must be one of'\n        ' [reg:squarederror, reg:squaredlogerror, reg:logistic,\
           \ reg:gamma,'\n        ' reg:tweedie, reg:pseudohubererror, binary:logistic,'\n\
           \        ' multi:softprob].'\n    )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-model-batch-predict:
       container:
         args:
@@ -3407,7 +3407,7 @@ deploymentSpec:
           \ 'data_source_csv_filenames',\n          'data_source_bigquery_table_path',\n\
           \      ],\n  )(\n      data_source_csv_filenames,\n      data_source_bigquery_table_path,\n\
           \  )\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/kfp-v2-base:20250827_0525
     exec-split-materialized-data:
       container:
         args:
@@ -3453,7 +3453,7 @@ deploymentSpec:
           \ 'w') as f:\n    f.write(file_patterns[0])\n\n  with tf.io.gfile.GFile(materialized_eval_split,\
           \ 'w') as f:\n    f.write(file_patterns[1])\n\n  with tf.io.gfile.GFile(materialized_test_split,\
           \ 'w') as f:\n    f.write(file_patterns[2])\n\n"
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/dataflow-worker:20250827_0525
     exec-training-configurator-and-validator:
       container:
         args:
@@ -3498,7 +3498,7 @@ deploymentSpec:
           ["--temporal_total_weight=", "{{$.inputs.parameters[''temporal_total_weight'']}}"]}}}'
         - '{"IfPresent": {"InputName": "group_temporal_total_weight", "Then": {"Concat":
           ["--group_temporal_total_weight=", "{{$.inputs.parameters[''group_temporal_total_weight'']}}"]}}}'
-        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250129_0625
+        image: us-docker.pkg.dev/vertex-ai/automl-tabular/feature-transform-engine:20250827_0525
     exec-xgboost-trainer:
       container:
         args: