Source code for astronomer.providers.amazon.aws.sensors.batch

import warnings

from airflow.providers.amazon.aws.sensors.batch import BatchSensor




class BatchSensorAsync(BatchSensor):
    """
    This class is deprecated.
    Please use :class: `~airflow.providers.amazon.aws.sensors.batch.BatchSensor`.
    """

    is_deprecated = True
    post_deprecation_replacement = "from airflow.providers.amazon.aws.sensors.batch import BatchSensor"

    def __init__(self, *args, **kwargs) -> None:  # type: ignore[no-untyped-def]
        warnings.warn(
            (
                "This module is deprecated. "
                "Please use `airflow.providers.amazon.aws.sensors.batch.BatchSensor` "
                "and set deferrable to True instead."
            ),
            DeprecationWarning,
            stacklevel=2,
        )
        return super().__init__(*args, deferrable=True, **kwargs)