Skip to content

Commit ca6b37f

Browse files
committed
fix: type hint.
1 parent 492bac3 commit ca6b37f

File tree

4 files changed

+36
-36
lines changed

4 files changed

+36
-36
lines changed

turu-snowflake/src/turu/snowflake/async_connection.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -183,23 +183,23 @@ async def execute_map(
183183
@overload
184184
async def execute_map(
185185
self,
186-
row_type: Type[GenericNewPanderaDataFrameModel],
186+
row_type: Type[GenericNewPyArrowTable],
187187
operation: str,
188188
parameters: "Optional[Any]" = None,
189189
/,
190190
**options: Unpack[ExecuteOptions],
191-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
191+
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
192192
...
193193

194194
@overload
195195
async def execute_map(
196196
self,
197-
row_type: Type[GenericNewPyArrowTable],
197+
row_type: Type[GenericNewPanderaDataFrameModel],
198198
operation: str,
199199
parameters: "Optional[Any]" = None,
200200
/,
201201
**options: Unpack[ExecuteOptions],
202-
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
202+
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
203203
...
204204

205205
@override
@@ -267,23 +267,23 @@ async def executemany_map(
267267
@overload
268268
async def executemany_map(
269269
self,
270-
row_type: Type[GenericNewPanderaDataFrameModel],
270+
row_type: Type[GenericNewPyArrowTable],
271271
operation: str,
272272
seq_of_parameters: Sequence[Any],
273273
/,
274274
**options: Unpack[ExecuteOptions],
275-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
275+
) -> AsyncCursor[Never, Never, GenericNewPyArrowTable]:
276276
...
277277

278278
@overload
279279
async def executemany_map(
280280
self,
281-
row_type: Type[GenericNewPyArrowTable],
281+
row_type: Type[GenericNewPanderaDataFrameModel],
282282
operation: str,
283283
seq_of_parameters: Sequence[Any],
284284
/,
285285
**options: Unpack[ExecuteOptions],
286-
) -> AsyncCursor[Never, Never, GenericNewPyArrowTable]:
286+
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
287287
...
288288

289289
@override

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

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -151,23 +151,23 @@ async def execute_map(
151151
@overload
152152
async def execute_map(
153153
self,
154-
row_type: Type[GenericNewPanderaDataFrameModel],
154+
row_type: Type[GenericNewPyArrowTable],
155155
operation: str,
156156
parameters: "Optional[Any]" = None,
157157
/,
158158
**options: Unpack[ExecuteOptions],
159-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
159+
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
160160
...
161161

162162
@overload
163163
async def execute_map(
164164
self,
165-
row_type: Type[GenericNewPyArrowTable],
165+
row_type: Type[GenericNewPanderaDataFrameModel],
166166
operation: str,
167167
parameters: "Optional[Any]" = None,
168168
/,
169169
**options: Unpack[ExecuteOptions],
170-
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
170+
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
171171
...
172172

173173
@override
@@ -227,23 +227,23 @@ async def executemany_map(
227227
@overload
228228
async def executemany_map(
229229
self,
230-
row_type: Type[GenericNewPanderaDataFrameModel],
230+
row_type: Type[GenericNewPyArrowTable],
231231
operation: str,
232232
seq_of_parameters: "Sequence[Any]",
233233
/,
234234
**options: Unpack[ExecuteOptions],
235-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
235+
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
236236
...
237237

238238
@overload
239239
async def executemany_map(
240240
self,
241-
row_type: Type[GenericNewPyArrowTable],
241+
row_type: Type[GenericNewPanderaDataFrameModel],
242242
operation: str,
243243
seq_of_parameters: "Sequence[Any]",
244244
/,
245245
**options: Unpack[ExecuteOptions],
246-
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
246+
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
247247
...
248248

249249
@override

turu-snowflake/src/turu/snowflake/connection.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -193,23 +193,23 @@ def execute_map(
193193
@overload
194194
def execute_map(
195195
self,
196-
row_type: Type[GenericNewPanderaDataFrameModel],
196+
row_type: Type[GenericNewPyArrowTable],
197197
operation: str,
198198
parameters: "Optional[Any]" = None,
199199
/,
200200
**options: Unpack[ExecuteOptions],
201-
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
201+
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
202202
...
203203

204204
@overload
205205
def execute_map(
206206
self,
207-
row_type: Type[GenericNewPyArrowTable],
207+
row_type: Type[GenericNewPanderaDataFrameModel],
208208
operation: str,
209209
parameters: "Optional[Any]" = None,
210210
/,
211211
**options: Unpack[ExecuteOptions],
212-
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
212+
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
213213
...
214214

215215
@override
@@ -277,23 +277,23 @@ def executemany_map(
277277
@overload
278278
def executemany_map(
279279
self,
280-
row_type: Type[GenericNewPanderaDataFrameModel],
280+
row_type: Type[GenericNewPyArrowTable],
281281
operation: str,
282282
seq_of_parameters: Sequence[Any],
283283
/,
284284
**options: Unpack[ExecuteOptions],
285-
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
285+
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
286286
...
287287

288288
@overload
289289
def executemany_map(
290290
self,
291-
row_type: Type[GenericNewPyArrowTable],
291+
row_type: Type[GenericNewPanderaDataFrameModel],
292292
operation: str,
293293
seq_of_parameters: Sequence[Any],
294294
/,
295295
**options: Unpack[ExecuteOptions],
296-
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
296+
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
297297
...
298298

299299
@override

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

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -149,23 +149,23 @@ def execute_map(
149149
@overload
150150
def execute_map(
151151
self,
152-
row_type: Type[GenericNewPyArrowTable],
152+
row_type: Type[GenericNewPanderaDataFrameModel],
153153
operation: str,
154154
parameters: "Optional[Any]" = None,
155155
/,
156156
**options: Unpack[ExecuteOptions],
157-
) -> "Cursor[Never, Never, GenericNewPyArrowTable]":
157+
) -> "Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
158158
...
159159

160160
@overload
161161
def execute_map(
162162
self,
163-
row_type: Type[GenericNewPanderaDataFrameModel],
163+
row_type: Type[GenericNewPyArrowTable],
164164
operation: str,
165165
parameters: "Optional[Any]" = None,
166166
/,
167167
**options: Unpack[ExecuteOptions],
168-
) -> "Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
168+
) -> "Cursor[Never, Never, GenericNewPyArrowTable]":
169169
...
170170

171171
@override
@@ -225,24 +225,24 @@ def executemany_map(
225225
@overload
226226
def executemany_map(
227227
self,
228-
row_type: Type[GenericNewPanderaDataFrameModel],
228+
row_type: Type[GenericNewPyArrowTable],
229229
operation: str,
230-
seq_of_parameters: Sequence[Any],
230+
seq_of_parameters: "Sequence[Any]",
231231
/,
232232
**options: Unpack[ExecuteOptions],
233-
) -> "Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
234-
...
233+
) -> "Cursor[Never, Never, GenericNewPyArrowTable]":
234+
pass
235235

236236
@overload
237237
def executemany_map(
238238
self,
239-
row_type: Type[GenericNewPyArrowTable],
239+
row_type: Type[GenericNewPanderaDataFrameModel],
240240
operation: str,
241-
seq_of_parameters: "Sequence[Any]",
241+
seq_of_parameters: Sequence[Any],
242242
/,
243243
**options: Unpack[ExecuteOptions],
244-
) -> "Cursor[Never, Never, GenericNewPyArrowTable]":
245-
pass
244+
) -> "Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
245+
...
246246

247247
@override
248248
def executemany_map(

0 commit comments

Comments
 (0)