Skip to content

Feature: Timeout Vertex AI Step#326

Open
haroldrubio wants to merge 6 commits intomasterfrom
feature/vertex-timeout
Open

Feature: Timeout Vertex AI Step#326
haroldrubio wants to merge 6 commits intomasterfrom
feature/vertex-timeout

Conversation

@haroldrubio
Copy link
Member

This PR adds a timeout to the score computation step of the Vertex AI Pipeline. The timeout cannot be set at runtime (e.g passed as a parameter per request at create_job) because the timeout is a compiled property of that pipeline step.

Tested by deploying to the dev site and checking that jobs fail when the config is set to 1 second (Branch: feature/vertex-timeout-1s)

@haroldrubio haroldrubio self-assigned this Jan 15, 2026
@haroldrubio
Copy link
Member Author

I'm unsure how to add a PyTest for this since we don't have tests to assert the structure of the compiled pipeline - but I could look into adding them

@haroldrubio haroldrubio marked this pull request as ready for review January 15, 2026 19:20

# Create a temporary config file
config_content = """
PIPELINE_TIMEOUT_SMALL = '86400s'
Copy link
Member

Choose a reason for hiding this comment

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

can you set a timeout of 1 second, run a job and expect to receive an error?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants