diff --git a/airflow/contrib/operators/bigquery_check_operator.py b/airflow/contrib/operators/bigquery_check_operator.py index 355929135283e7..c34fd52f7083e2 100644 --- a/airflow/contrib/operators/bigquery_check_operator.py +++ b/airflow/contrib/operators/bigquery_check_operator.py @@ -18,12 +18,12 @@ # under the License. from airflow.contrib.hooks.bigquery_hook import BigQueryHook -from airflow.operators.check_operator import \ - CheckOperator, ValueCheckOperator, IntervalCheckOperator +from airflow.operators.sql import \ + SQLCheckOperator, SQLValueCheckOperator, SQLIntervalCheckOperator from airflow.utils.decorators import apply_defaults -class BigQueryCheckOperator(CheckOperator): +class BigQueryCheckOperator(SQLCheckOperator): """ Performs checks against BigQuery. The ``BigQueryCheckOperator`` expects a sql query that will return a single row. Each value on that @@ -79,7 +79,7 @@ def get_db_hook(self): use_legacy_sql=self.use_legacy_sql) -class BigQueryValueCheckOperator(ValueCheckOperator): +class BigQueryValueCheckOperator(SQLValueCheckOperator): """ Performs a simple value check using sql code. @@ -111,7 +111,7 @@ def get_db_hook(self): use_legacy_sql=self.use_legacy_sql) -class BigQueryIntervalCheckOperator(IntervalCheckOperator): +class BigQueryIntervalCheckOperator(SQLIntervalCheckOperator): """ Checks that the values of metrics given as SQL expressions are within a certain tolerance of the ones from days_back before.