Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix parent_model parameter in GCP Vertex AI AutoML and Custom Job operators #38417

Merged
merged 1 commit into from
Mar 23, 2024

Conversation

shahar1
Copy link
Contributor

@shahar1 shahar1 commented Mar 22, 2024

closes: #37400

The following line wrongly parsed the parent_model param in all Vertex AI AutoML "Create" operators:

self.parent_model = self.parent_model.rpartition("@")[0] if self.parent_model else None

The parent_model should be either the actualy ID (integer) or the its resource name format (e.g., projects/my-project/location/my-location/models/parent_model_id). After removing the line, it seems to work fine (I checked it).


^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

@boring-cyborg boring-cyborg bot added area:providers provider:google Google (including GCP) related issues labels Mar 22, 2024
@shahar1 shahar1 force-pushed the fix-automl-parent-model-parsing branch from cd102bb to c249af4 Compare March 22, 2024 20:57
Copy link
Collaborator

@dirrao dirrao left a comment

Choose a reason for hiding this comment

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

LGTM

@shahar1 shahar1 force-pushed the fix-automl-parent-model-parsing branch from c249af4 to 108318c Compare March 23, 2024 11:34
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.

Vertex AI - model versioning doesn't work with CreateAutoMLTextTrainingJobOperator
3 participants