potiuk commented on code in PR #31118:
URL: https://github.com/apache/airflow/pull/31118#discussion_r1188142923
##########
airflow/providers/google/cloud/hooks/vertex_ai/batch_prediction_job.py:
##########
@@ -207,6 +209,14 @@ def create_batch_prediction_job(
:param sync: Whether to execute this method synchronously. If False,
this method will be executed in
concurrent Future and any downstream object will be immediately
returned and synced when the
Future has completed.
+ :param create_request_timeout: Optional. The timeout for the create
request in seconds.
+ :param batch_size: Optional. The number of the records (e.g. instances)
+ of the operation given in each batch
+ to a machine replica. Machine type, and size of a single record
should be considered
+ when setting this parameter, higher value speeds up the batch
operation's execution,
+ but too high value will result in a whole batch not fitting in a
machine's memory,
+ and the whole operation will fail.
+ The default value is 64.
Review Comment:
```suggestion
The default value is same as in the aiplatform's
BatchPredictionJob.
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
