Better exception raised in case of numpy missing (apache#28722)

We should also raise optional feature error when numpy is missing
Jorricks · Jan 4, 2023 · 8c5ee5e · 8c5ee5e
1 parent 93fed0c
commit 8c5ee5e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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