diff --git a/pyathena/async_cursor.py b/pyathena/async_cursor.py index d9f40f73..64a95537 100644 --- a/pyathena/async_cursor.py +++ b/pyathena/async_cursor.py @@ -97,7 +97,7 @@ def _collect_result_set(self, query_id: str) -> AthenaResultSet: def execute( self, operation: str, - parameters: Optional[Dict[str, Any]] = None, + parameters: Optional[Union[Dict[str, Any], List[str]]] = None, work_group: Optional[str] = None, s3_staging_dir: Optional[str] = None, cache_size: Optional[int] = 0, diff --git a/pyathena/pandas/cursor.py b/pyathena/pandas/cursor.py index 5c273ab3..0d39f8e3 100644 --- a/pyathena/pandas/cursor.py +++ b/pyathena/pandas/cursor.py @@ -182,7 +182,10 @@ def execute( return self def executemany( - self, operation: str, seq_of_parameters: List[Optional[Dict[str, Any]]], **kwargs + self, + operation: str, + seq_of_parameters: List[Optional[Union[Dict[str, Any], List[str]]]], + **kwargs, ) -> None: for parameters in seq_of_parameters: self.execute(operation, parameters, **kwargs) diff --git a/pyathena/spark/common.py b/pyathena/spark/common.py index 8d67cfd6..c85be14b 100644 --- a/pyathena/spark/common.py +++ b/pyathena/spark/common.py @@ -217,7 +217,10 @@ def close(self) -> None: self._terminate_session() def executemany( - self, operation: str, seq_of_parameters: List[Optional[Dict[str, Any]]], **kwargs + self, + operation: str, + seq_of_parameters: List[Optional[Union[Dict[str, Any], List[str]]]], + **kwargs, ) -> None: raise NotSupportedError diff --git a/pyathena/sqlalchemy/util.py b/pyathena/sqlalchemy/util.py index 3f345e35..f8c47f7e 100644 --- a/pyathena/sqlalchemy/util.py +++ b/pyathena/sqlalchemy/util.py @@ -2,5 +2,5 @@ class _HashableDict(dict): # type: ignore - def __hash__(self): + def __hash__(self): # type: ignore return hash(tuple(sorted(self.items())))