-
Notifications
You must be signed in to change notification settings - Fork 350
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
feat: Added create_training_pipeline_custom_job_sample and create_training_pipeline_custom_training_managed_dataset_sample and fixed create_training_pipeline_image_classification_sample #343
Conversation
chore: merge main into dev
Here is the summary of changes. You are about to add 2 region tags.
This comment is generated by snippet-bot.
|
85fff2d
to
0ca06d5
Compare
… and fixed unmanaged sample
4ea56bf
to
68eff56
Compare
|
||
|
||
# [START aiplatform_sdk_create_and_import_dataset_tabular_bigquery_sample] | ||
def create_and_import_dataset_tabular_bigquery_sample( |
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.
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.
Yes, but it's under my name. Let me check with @aribray.
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.
The assignments might have changed between her writing the sample and me starting. I will remove these and we can use Ari's.
def create_training_pipeline_custom_job_sample( | ||
project: str, | ||
display_name: str, | ||
args: List[str], |
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.
args
are optional and should default to None
.
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.
Done.
script_path: str, | ||
container_uri: str, | ||
location: str = "us-central1", | ||
model_display_name: str = None, |
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: str = None, | |
model_display_name: Optional[str] = None, |
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.
Could do it, but I followed the example of the existing create_training_pipeline_image_classification_sample
. I will change it there too.
script_path: str, | ||
container_uri: str, | ||
location: str = "us-central1", | ||
model_display_name: str = None, |
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.
We should also include the model_serving_container_image_uri
as required since we are returning the Model in the sample.
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.
Done.
container_uri: str, | ||
dataset_id: int, | ||
location: str = "us-central1", | ||
model_display_name: str = None, |
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.
Same comments here as above.
@sasha-gitg I fixed a couple other things along the same lines, such as order of args. |
22abd8c
to
fec3e6c
Compare
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.
Please fix linting errors!
@dinagraves Linting fixed. Can I get another look? I usually run the linter after all the other checks/approvals pass so I only run it once. |
Added 2 samples with tests:
Fixed: