Skip to content

Add reserved IP ranges to Vertex AI pipeline jobs - #72560

Merged
potiuk merged 2 commits into
apache:mainfrom
LE0-Lin:add-vertex-ai-reserved-ip-ranges
Sep 9, 2026
Merged

potiuk merged 2 commits into
apache:mainfrom
LE0-Lin:add-vertex-ai-reserved-ip-ranges

Conversation

@LE0-Lin

@LE0-Lin LE0-Lin commented Sep 5, 2026 •

Copy link
Copy Markdown
Contributor

Adds reserved_ip_ranges support to RunPipelineJobOperator so Vertex AI pipeline jobs can be restricted to configured VPC IP ranges. The value is propagated through both hook submission paths without collapsing None or an empty list, with unit coverage at the operator and SDK boundaries.

Tests:

  • pytest providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_pipeline_job.py (25 passed)
  • pytest providers/google/tests/unit/google/cloud/operators/test_vertex_ai.py -k RunPipelineJobOperator (6 passed)
  • Google provider prek checks, Ruff formatting/lint, and git diff --check

closes: #62733


Was generative AI tooling used to co-author this PR?
  • Yes — OpenAI Codex

Generated-by: OpenAI Codex following the guidelines

OpenAI Codex was used for limited implementation and testing assistance. All changes were reviewed and validated by me.


Drafted-by: OpenAI Codex (reviewed by @LE0-Lin)

@LE0-Lin
LE0-Lin requested a review from shahar1 as a code owner September 5, 2026 11:15
@boring-cyborg boring-cyborg Bot added area:providers provider:google Google (including GCP) related issues labels Sep 5, 2026

@SameerMesiah97 SameerMesiah97 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The test could benefit from a slight adjustment. Looks good otherwise.

Comment thread providers/google/tests/unit/google/cloud/hooks/vertex_ai/test_pipeline_job.py Outdated
@LE0-Lin
LE0-Lin force-pushed the add-vertex-ai-reserved-ip-ranges branch from 7e4124e to 2898959 Compare September 5, 2026 16:49

@potiuk potiuk left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@potiuk
potiuk merged commit 9246fc5 into apache:main Sep 9, 2026
86 checks passed
@boring-cyborg

boring-cyborg Bot commented Sep 9, 2026

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Sep 11, 2026
* Add reserved IP ranges to Vertex AI pipeline jobs

* Split Vertex AI pipeline job hook tests
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:providers provider:google Google (including GCP) related issues

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Include reserved_ip_ranges to airflow.providers.google.cloud.operators.vertex_ai.pipeline_job.RunPipelineJobTrigger

3 participants