Change to lazy loading of io in pandas serializer

apache · Sep 29, 2023 · b774d31 · b774d31
1 parent 55fbdfe
commit b774d31
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/serialization/serializers/pandas.py b/airflow/serialization/serializers/pandas.py
@@ -17,7 +17,6 @@
 # under the License.
 from __future__ import annotations
 
-from io import BytesIO
 from typing import TYPE_CHECKING
 
 from airflow.utils.module_loading import qualname
@@ -63,6 +62,8 @@ def deserialize(classname: str, version: int, data: object) -> pd.DataFrame:
     if not isinstance(data, str):
         raise TypeError(f"serialized {classname} has wrong data type {type(data)}")
 
+    from io import BytesIO
+
     with BytesIO(bytes.fromhex(data)) as buf:
         df = pq.read_table(buf).to_pandas()