From 0f5562e7634c8e38b47793e7c744d29116f2e9e7 Mon Sep 17 00:00:00 2001 From: Ankit Chaurasia <8670962+sunank200@users.noreply.github.com> Date: Tue, 1 Mar 2022 02:35:25 +0545 Subject: [PATCH] Fix mypy issues for bigquery async --- astronomer/providers/google/cloud/hooks/bigquery.py | 2 +- astronomer/providers/google/cloud/triggers/bigquery.py | 2 +- astronomer/providers/google/cloud/triggers/gcs.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/astronomer/providers/google/cloud/hooks/bigquery.py b/astronomer/providers/google/cloud/hooks/bigquery.py index 014213168..8a8cf32fc 100644 --- a/astronomer/providers/google/cloud/hooks/bigquery.py +++ b/astronomer/providers/google/cloud/hooks/bigquery.py @@ -203,7 +203,7 @@ def value_check( raise AirflowException(error_msg) @staticmethod - def _get_numeric_matches(records: list[float], pass_value: float, tolerance: float = None): + def _get_numeric_matches(records, pass_value: float, tolerance: float = None): """ A helper function to match numeric pass_value, tolerance with records value diff --git a/astronomer/providers/google/cloud/triggers/bigquery.py b/astronomer/providers/google/cloud/triggers/bigquery.py index e35a62d14..120ef7ac9 100644 --- a/astronomer/providers/google/cloud/triggers/bigquery.py +++ b/astronomer/providers/google/cloud/triggers/bigquery.py @@ -200,7 +200,7 @@ def __init__( second_job_id: str, project_id: Optional[str], table: str, - metrics_thresholds: dict[str, Any], + metrics_thresholds: Dict[str, Any], date_filter_column: Optional[str] = "ds", days_back: SupportsAbs[int] = -7, ratio_formula: str = "max_over_min", diff --git a/astronomer/providers/google/cloud/triggers/gcs.py b/astronomer/providers/google/cloud/triggers/gcs.py index 3cd1f3746..225542890 100644 --- a/astronomer/providers/google/cloud/triggers/gcs.py +++ b/astronomer/providers/google/cloud/triggers/gcs.py @@ -26,7 +26,7 @@ def __init__( object_name: str, polling_period_seconds: float, google_cloud_conn_id: str, - hook_params: dict[str, Any], + hook_params: Dict[str, Any], ): self.bucket = bucket self.object_name = object_name