ferruzzi commented on code in PR #48468:
URL: https://github.com/apache/airflow/pull/48468#discussion_r2019054649
##########
providers/amazon/docs/operators/bedrock.rst:
##########
@@ -298,6 +329,32 @@ To wait on the state of an Amazon Bedrock data ingestion
job until it reaches a
:start-after: [START howto_sensor_bedrock_ingest_data]
:end-before: [END howto_sensor_bedrock_ingest_data]
+.. _howto/sensor:BedrockBatchInferenceJobScheduledSensor:
+
+Wait for an Amazon Bedrock batch inference job to be Scheduled
+===============================================================
+
+To wait on the state of an Amazon Bedrock batch inference job until it reaches
the "Scheduled" state you can use
+:class:`~airflow.providers.amazon.aws.sensors.bedrock.BedrockBatchInferenceScheduledSensor`
+
+.. exampleinclude::
/../../providers/amazon/tests/system/amazon/aws/example_bedrock_batch_inference.py
+ :language: python
+ :dedent: 4
+ :start-after: [START howto_sensor_bedrock_batch_inference_scheduled]
+ :end-before: [END howto_sensor_bedrock_batch_inference_scheduled]
+
+.. _howto/sensor:BedrockBatchInferenceJobCompleteSensor:
+
+Wait for an Amazon Bedrock batch inference job to be finished
Review Comment:
I suppose. Would you rather one sensor that takes a parameter to pick the
desired end state? If so, should it default to waiting for completion and
optionally skip out once it is scheduled?
--
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]
