Merge pull request #86 from yassun7010/add_typehint

chore: add type hint.

Author: ya7010

turu-snowflake/src/turu/snowflake/async_cursor.py

@@ -318,7 +318,7 @@ async def fetch_arrow_batches(self) -> AsyncIterator[GenericPyArrowTable]:
         for batch in self._raw_cursor.fetch_arrow_batches():
             yield cast(GenericPyArrowTable, batch)
 
-    async def fetch_pandas_all(self, **kwargs) -> GenericPandasDataFrame:
+    async def fetch_pandas_all(self, **kwargs: Any) -> GenericPandasDataFrame:
         """Fetch Pandas dataframes."""
 
         df = self._raw_cursor.fetch_pandas_all(**kwargs)
@@ -329,7 +329,7 @@ async def fetch_pandas_all(self, **kwargs) -> GenericPandasDataFrame:
         return cast(GenericPandasDataFrame, df)
 
     async def fetch_pandas_batches(
-        self, **kwargs
+        self, **kwargs: Any
     ) -> AsyncIterator[GenericPandasDataFrame]:
         """Fetch Pandas dataframes in batches, where 'batch' refers to Snowflake Chunk."""
 

turu-snowflake/src/turu/snowflake/cursor.py

@@ -331,7 +331,7 @@ def fetch_arrow_batches(self) -> "Iterator[GenericPyArrowTable]":
             Iterator[GenericPyArrowTable], self._raw_cursor.fetch_arrow_batches()
         )
 
-    def fetch_pandas_all(self, **kwargs) -> "GenericPandasDataFrame":
+    def fetch_pandas_all(self, **kwargs: Any) -> "GenericPandasDataFrame":
         """Fetch a single Pandas dataframe."""
         df = self._raw_cursor.fetch_pandas_all(**kwargs)
 
@@ -340,7 +340,7 @@ def fetch_pandas_all(self, **kwargs) -> "GenericPandasDataFrame":
 
         return cast(GenericPandasDataFrame, df)
 
-    def fetch_pandas_batches(self, **kwargs) -> "Iterator[GenericPandasDataFrame]":
+    def fetch_pandas_batches(self, **kwargs: Any) -> "Iterator[GenericPandasDataFrame]":
         """Fetch Pandas dataframes in batches, where 'batch' refers to Snowflake Chunk."""
 
         return cast(