Description
Using Vertex AI, it is possible to upload a model as a new version of an existing model. This is possible by passing the parent_model to the UploadModelRequest. However, the ModelServiceHook.upload_model (under airflow.providers.google.cloud.hooks.vertex_ai.model_service.py) does not allow a user to pass the parent_model.
Currently, when a new model is uploaded, a new model is created instead of creating a new version on an already existing model.
Use case/motivation
Adding this feature enables users to use the model versioning functionality in Vertex AI.
Related issues
No response
Are you willing to submit a PR?
Code of Conduct
Description
Using Vertex AI, it is possible to upload a model as a new version of an existing model. This is possible by passing the
parent_modelto theUploadModelRequest. However, theModelServiceHook.upload_model(underairflow.providers.google.cloud.hooks.vertex_ai.model_service.py) does not allow a user to pass theparent_model.Currently, when a new model is uploaded, a new model is created instead of creating a new version on an already existing model.
Use case/motivation
Adding this feature enables users to use the model versioning functionality in Vertex AI.
Related issues
No response
Are you willing to submit a PR?
Code of Conduct