Skip to content

Commit d4fa086

Browse files
committed
fix: type.
1 parent ca6b37f commit d4fa086

File tree

2 files changed

+6
-16
lines changed

2 files changed

+6
-16
lines changed

turu-snowflake/src/turu/snowflake/mock_async_connection.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -70,16 +70,6 @@ def inject_response(
7070
) -> Self:
7171
...
7272

73-
@overload
74-
def inject_response(
75-
self,
76-
row_type: Type[GenericPanderaDataFrameModel],
77-
response: Union[
78-
Sequence[GenericPandasDataFrame], GenericPandasDataFrame, Exception
79-
],
80-
) -> Self:
81-
...
82-
8373
@overload
8474
def inject_response(
8575
self,

turu-snowflake/src/turu/snowflake/mock_connection.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -73,18 +73,18 @@ def inject_response(
7373
@overload
7474
def inject_response(
7575
self,
76-
row_type: Type[GenericPanderaDataFrameModel],
77-
response: Union[
78-
Sequence[GenericPandasDataFrame], GenericPandasDataFrame, Exception
79-
],
76+
row_type: Type[GenericPyArrowTable],
77+
response: Union[Sequence[GenericPyArrowTable], GenericPyArrowTable, Exception],
8078
) -> Self:
8179
...
8280

8381
@overload
8482
def inject_response(
8583
self,
86-
row_type: Type[GenericPyArrowTable],
87-
response: Union[Sequence[GenericPyArrowTable], GenericPyArrowTable, Exception],
84+
row_type: Type[GenericPanderaDataFrameModel],
85+
response: Union[
86+
Sequence[GenericPandasDataFrame], GenericPandasDataFrame, Exception
87+
],
8888
) -> Self:
8989
...
9090

0 commit comments

Comments
 (0)