Fix F821 Undefined name List and Union

laughingman7743 · Dec 28, 2024 · 1ea93bd · 1ea93bd
1 parent 7abcca7
commit 1ea93bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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