Skip to content

Add Workflow to filter spot zones#5548

Merged
AdarshK15 merged 6 commits into
GoogleCloudPlatform:developfrom
AdarshK15:filter-spot-zones
Apr 24, 2026
Merged

Add Workflow to filter spot zones#5548
AdarshK15 merged 6 commits into
GoogleCloudPlatform:developfrom
AdarshK15:filter-spot-zones

Conversation

@AdarshK15

Copy link
Copy Markdown
Member

Summary

The current workflow just creates list of zones with available capacity for accelerator types, but doesn't filter zones which doesn't have other required infrastructure dependencies.

Adds tools/cloud-build/filter-zone-options.yaml to dynamically update integration test options files in GCS.

The workflow:

  • Scans test definitions and blueprints for infrastructure dependencies (Filestore, Managed Lustre, CPU tiers).
  • Checks GCP for compatible locations.
  • Rewrites the options files to drop zones that lack the required resources.
    This prevents deployments from failing due to unsupported features in the target zone.

@gemini-code-assist

Copy link
Copy Markdown
Contributor

Summary of Changes

Hello, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request introduces a new automated workflow designed to improve the reliability of integration tests by ensuring that target GCP zones support the necessary infrastructure dependencies. By dynamically scanning test configurations and verifying resource availability against GCP APIs, the workflow proactively filters out incompatible zones, preventing deployment failures caused by missing infrastructure features.

Highlights

  • New Workflow Implementation: Added a new cloud build workflow script to dynamically filter GCP zones based on infrastructure requirements.
  • Dependency Awareness: The workflow now scans test definitions and blueprints to identify dependencies on Filestore, Managed Lustre, and specific CPU machine types.
  • Automated GCS Updates: Automatically updates integration test options files in GCS by removing zones that do not support the required infrastructure.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for GitHub and other Google products, sign up here.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

@gemini-code-assist gemini-code-assist Bot 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.

Code Review

This pull request introduces a new Cloud Build configuration, filter-zone-options.yaml, which automates the filtering of GCP zones for daily tests based on resource requirements such as Filestore, Managed Lustre, and specific CPU machine types. The review identifies a critical bug in the Filestore location retrieval format and suggests several improvements, including broadening the CPU machine type regex to support more machine families, parameterizing the hardcoded GCS bucket for better flexibility, and extracting the extensive inline shell script into a separate file to improve maintainability.

Comment thread tools/cloud-build/filter-zone-options.yaml Outdated
Comment thread tools/cloud-build/filter-zone-options.yaml Outdated
Comment thread tools/cloud-build/filter-zone-options.yaml Outdated
Comment thread tools/cloud-build/filter-zone-options.yaml
LAVEEN
LAVEEN previously approved these changes Apr 23, 2026
@AdarshK15 AdarshK15 marked this pull request as ready for review April 24, 2026 13:31
@AdarshK15 AdarshK15 requested a review from a team as a code owner April 24, 2026 13:31
@AdarshK15 AdarshK15 added the release-chore To not include into release notes label Apr 24, 2026
@AdarshK15 AdarshK15 merged commit 81d452f into GoogleCloudPlatform:develop Apr 24, 2026
16 of 77 checks passed
@AdarshK15 AdarshK15 deleted the filter-spot-zones branch May 3, 2026 18:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

release-chore To not include into release notes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants