From 961ea033e5213ca928ce84fb5b11712935e007a1 Mon Sep 17 00:00:00 2001 From: Jarek Potiuk Date: Wed, 4 Jan 2023 10:58:23 +0100 Subject: [PATCH] Better exception raised in case of numpy missing We should also raise optional feature error when numpy is missing --- airflow/providers/amazon/aws/transfers/sql_to_s3.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/airflow/providers/amazon/aws/transfers/sql_to_s3.py b/airflow/providers/amazon/aws/transfers/sql_to_s3.py index 1ffc736ba3f29..d10f781f5453a 100644 --- a/airflow/providers/amazon/aws/transfers/sql_to_s3.py +++ b/airflow/providers/amazon/aws/transfers/sql_to_s3.py @@ -22,9 +22,8 @@ from tempfile import NamedTemporaryFile from typing import TYPE_CHECKING, Iterable, Mapping, Sequence -import numpy as np - try: + import numpy as np import pandas as pd except ImportError as e: from airflow.exceptions import AirflowOptionalProviderFeatureException