diff --git a/pyathena/pandas/async_cursor.py b/pyathena/pandas/async_cursor.py index bea6677f..d6317e04 100644 --- a/pyathena/pandas/async_cursor.py +++ b/pyathena/pandas/async_cursor.py @@ -4,7 +4,7 @@ import logging from concurrent.futures import Future from multiprocessing import cpu_count -from typing import Any, Dict, Iterable, Optional, Tuple, Union, cast +from typing import Any, Dict, Iterable, List, Optional, Tuple, Union, cast from pyathena import ProgrammingError from pyathena.async_cursor import AsyncCursor diff --git a/pyathena/spark/cursor.py b/pyathena/spark/cursor.py index 63bfe546..3bf46375 100644 --- a/pyathena/spark/cursor.py +++ b/pyathena/spark/cursor.py @@ -2,7 +2,7 @@ from __future__ import annotations import logging -from typing import Any, Dict, Optional, cast +from typing import Any, Dict, List, Optional, Union, cast from pyathena import OperationalError, ProgrammingError from pyathena.model import AthenaCalculationExecution, AthenaCalculationExecutionStatus