This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git.
from d91b833 SalesforceToS3Operator: update sample dag and doc (#22489)
add ca4b8d1 Create Endpoint and Model Service, Batch Prediction and
Hyperparameter Tuning Jobs operators for Vertex AI service (#22088)
No new revisions were added by this update.
Summary of changes:
.../google/cloud/example_dags/example_vertex_ai.py | 292 ++++++++++
.../cloud/hooks/vertex_ai/batch_prediction_job.py | 347 ++++++++++++
.../cloud/hooks/vertex_ai/endpoint_service.py | 372 +++++++++++++
.../hooks/vertex_ai/hyperparameter_tuning_job.py | 415 ++++++++++++++
.../google/cloud/hooks/vertex_ai/model_service.py | 225 ++++++++
airflow/providers/google/cloud/links/vertex_ai.py | 206 ++++++-
.../operators/vertex_ai/batch_prediction_job.py | 510 +++++++++++++++++
.../vertex_ai/{dataset.py => endpoint_service.py} | 393 +++++++------
.../vertex_ai/hyperparameter_tuning_job.py | 504 +++++++++++++++++
.../cloud/operators/vertex_ai/model_service.py | 373 +++++++++++++
airflow/providers/google/provider.yaml | 16 +
.../operators/cloud/vertex_ai.rst | 160 ++++++
.../hooks/vertex_ai/test_batch_prediction_job.py | 184 ++++++
.../cloud/hooks/vertex_ai/test_endpoint_service.py | 377 +++++++++++++
.../vertex_ai/test_hyperparameter_tuning_job.py | 185 +++++++
.../cloud/hooks/vertex_ai/test_model_service.py | 214 +++++++
.../google/cloud/operators/test_vertex_ai.py | 616 ++++++++++++++++++++-
.../cloud/operators/test_vertex_ai_system.py | 16 +
18 files changed, 5239 insertions(+), 166 deletions(-)
create mode 100644
airflow/providers/google/cloud/hooks/vertex_ai/batch_prediction_job.py
create mode 100644
airflow/providers/google/cloud/hooks/vertex_ai/endpoint_service.py
create mode 100644
airflow/providers/google/cloud/hooks/vertex_ai/hyperparameter_tuning_job.py
create mode 100644
airflow/providers/google/cloud/hooks/vertex_ai/model_service.py
create mode 100644
airflow/providers/google/cloud/operators/vertex_ai/batch_prediction_job.py
copy airflow/providers/google/cloud/operators/vertex_ai/{dataset.py =>
endpoint_service.py} (65%)
create mode 100644
airflow/providers/google/cloud/operators/vertex_ai/hyperparameter_tuning_job.py
create mode 100644
airflow/providers/google/cloud/operators/vertex_ai/model_service.py
create mode 100644
tests/providers/google/cloud/hooks/vertex_ai/test_batch_prediction_job.py
create mode 100644
tests/providers/google/cloud/hooks/vertex_ai/test_endpoint_service.py
create mode 100644
tests/providers/google/cloud/hooks/vertex_ai/test_hyperparameter_tuning_job.py
create mode 100644
tests/providers/google/cloud/hooks/vertex_ai/test_model_service.py
