Skip to content

Feat: add batch_size kwarg for batch prediction jobs#1194

Merged
jaycee-li merged 7 commits intogoogleapis:mainfrom
jaycee-li:main
May 10, 2022
Merged

Feat: add batch_size kwarg for batch prediction jobs#1194
jaycee-li merged 7 commits intogoogleapis:mainfrom
jaycee-li:main

Conversation

@jaycee-li
Copy link
Contributor

@jaycee-li jaycee-li commented May 4, 2022

Fixes #525

Add batch_size kwarg for batch prediction jobs

@jaycee-li jaycee-li added the aiplatform Issues related to the AI Platform (Unified) API. label May 4, 2022
@product-auto-label product-auto-label bot added the size: m Pull request size is medium. label May 4, 2022
@jaycee-li jaycee-li requested a review from sasha-gitg May 4, 2022 17:16
@jaycee-li jaycee-li changed the title Add batch_size kwarg for batch prediction jobs Fix: add batch_size kwarg for batch prediction jobs May 4, 2022
@jaycee-li jaycee-li changed the title Fix: add batch_size kwarg for batch prediction jobs Feat: add batch_size kwarg for batch prediction jobs May 4, 2022
Copy link
Member

@sasha-gitg sasha-gitg left a comment

Choose a reason for hiding this comment

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

Mainly LGTM! Minor comments.

Update the copyright year. Change the order of the argument. Fix the syntax error.
@product-auto-label product-auto-label bot added the api: vertex-ai Issues related to the googleapis/python-aiplatform API. label May 5, 2022
@sararob sararob added the do not merge Indicates a pull request not ready for merge, due to either quality or timing. label May 5, 2022
@sararob sararob removed the do not merge Indicates a pull request not ready for merge, due to either quality or timing. label May 10, 2022
@jaycee-li jaycee-li merged commit 50bdb01 into googleapis:main May 10, 2022
@rlucioni
Copy link

@jaycee-li any chance we can get 1.13.1 pushed to PyPI? We'd love access to the batch_size kwarg. I see #1216 has been sitting for over a week.

@jaycee-li
Copy link
Contributor Author

@rlucioni I can ask the team to release 1.13.1 next week.

@rlucioni
Copy link

That'd be great, thanks!

@jaycee-li
Copy link
Contributor Author

@rlucioni
Hi Renzo, after discussing with the team, we decide not to release the patch version 1.13.1 since there is no urgent bug fix in this patch. We'll schedule to release 1.14.0 in two weeks.

The batch_size kwarg has been added to our main repo. So if you want to use it now, you can install from Github main branch: pip3 install --force-reinstall git+https://github.com/googleapis/python-aiplatform.git@main

Sorry for the inconvenience!

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

Labels

aiplatform Issues related to the AI Platform (Unified) API. api: vertex-ai Issues related to the googleapis/python-aiplatform API. size: m Pull request size is medium.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Support specifying batch_size when submitting batch predict jobs

4 participants