Skip to content

Commit

Permalink
Update base classes used in bigquery_check_operator to renamed versions
Browse files Browse the repository at this point in the history
  • Loading branch information
Dewald Abrie authored and kaxil committed Aug 14, 2020
1 parent 6a729c5 commit 73e1934
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions airflow/contrib/operators/bigquery_check_operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 73e1934

Please sign in to comment.