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: remove model_display_name requirement for training jobs #324
fix: remove model_display_name requirement for training jobs #324
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Thanks Morgan!
@@ -1491,6 +1491,9 @@ def _prepare_and_validate_run( | |||
""" | |||
) | |||
|
|||
if self._managed_model.container_spec.image_uri: | |||
model_display_name = model_display_name or self._display_name |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
model_display_name = model_display_name or self._display_name | |
model_display_name = model_display_name or self._display_name + '-model' |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Also can you update the docstring to capture this behavior.
6720044
to
9c7f1e9
Compare
Please open this PR into the main branch instead of dev. |
9c7f1e9
to
6cf7b83
Compare
…en model_serving_container_image_uri is provided
6cf7b83
to
72caf45
Compare
…en model_serving_container_image_uri is provided (googleapis#324)
Fixes #<b/184155454> 🦕