o-nikolas commented on code in PR #48468:
URL: https://github.com/apache/airflow/pull/48468#discussion_r2025197822
##########
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:
Left one comment about doc strings, otherwise lgtm!
--
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]
