Skip to content

Commit b5f6083

Browse files
committed
fix: type hint.
1 parent c5e5fe5 commit b5f6083

File tree

9 files changed

+107
-107
lines changed

9 files changed

+107
-107
lines changed

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

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -183,33 +183,33 @@ 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
206206
async def execute_map(
207207
self,
208208
row_type: Union[
209209
Type[GenericNewRowType],
210-
Type[GenericNewPanderaDataFrameModel],
211210
Type[GenericNewPandasDataFrame],
212211
Type[GenericNewPyArrowTable],
212+
Type[GenericNewPanderaDataFrameModel],
213213
],
214214
operation: str,
215215
parameters: Optional[Any] = None,
@@ -256,44 +256,44 @@ async def executemany_map(
256256
@overload
257257
async def executemany_map(
258258
self,
259-
row_type: Type[GenericNewPanderaDataFrameModel],
259+
row_type: Type[GenericNewPandasDataFrame],
260260
operation: str,
261261
seq_of_parameters: Sequence[Any],
262262
/,
263263
**options: Unpack[ExecuteOptions],
264-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
264+
) -> AsyncCursor[Never, GenericNewPandasDataFrame, Never]:
265265
...
266266

267267
@overload
268268
async def executemany_map(
269269
self,
270-
row_type: Type[GenericNewPandasDataFrame],
270+
row_type: Type[GenericNewPyArrowTable],
271271
operation: str,
272272
seq_of_parameters: Sequence[Any],
273273
/,
274274
**options: Unpack[ExecuteOptions],
275-
) -> AsyncCursor[Never, GenericNewPandasDataFrame, 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
290290
async def executemany_map(
291291
self,
292292
row_type: Union[
293293
Type[GenericNewRowType],
294-
Type[GenericNewPanderaDataFrameModel],
295294
Type[GenericNewPandasDataFrame],
296295
Type[GenericNewPyArrowTable],
296+
Type[GenericNewPanderaDataFrameModel],
297297
],
298298
operation: str,
299299
seq_of_parameters: Sequence[Any],

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

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -141,44 +141,44 @@ async def execute_map(
141141
@overload
142142
async def execute_map(
143143
self,
144-
row_type: Type[GenericNewPanderaDataFrameModel],
144+
row_type: Type[GenericNewPandasDataFrame],
145145
operation: str,
146146
parameters: "Optional[Any]" = None,
147147
/,
148148
**options: Unpack[ExecuteOptions],
149-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
149+
) -> "AsyncCursor[Never, GenericNewPandasDataFrame, Never]":
150150
...
151151

152152
@overload
153153
async def execute_map(
154154
self,
155-
row_type: Type[GenericNewPandasDataFrame],
155+
row_type: Type[GenericNewPyArrowTable],
156156
operation: str,
157157
parameters: "Optional[Any]" = None,
158158
/,
159159
**options: Unpack[ExecuteOptions],
160-
) -> "AsyncCursor[Never, GenericNewPandasDataFrame, Never]":
160+
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
161161
...
162162

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

174174
@override
175175
async def execute_map(
176176
self,
177177
row_type: Union[
178178
Type[GenericNewRowType],
179-
Type[GenericNewPanderaDataFrameModel],
180179
Type[GenericNewPandasDataFrame],
181180
Type[GenericNewPyArrowTable],
181+
Type[GenericNewPanderaDataFrameModel],
182182
],
183183
operation: str,
184184
parameters: "Optional[Any]" = None,
@@ -217,44 +217,44 @@ async def executemany_map(
217217
@overload
218218
async def executemany_map(
219219
self,
220-
row_type: Type[GenericNewPanderaDataFrameModel],
220+
row_type: Type[GenericNewPandasDataFrame],
221221
operation: str,
222222
seq_of_parameters: "Sequence[Any]",
223223
/,
224224
**options: Unpack[ExecuteOptions],
225-
) -> "AsyncCursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]":
225+
) -> "AsyncCursor[Never, GenericNewPandasDataFrame, Never]":
226226
...
227227

228228
@overload
229229
async def executemany_map(
230230
self,
231-
row_type: Type[GenericNewPandasDataFrame],
231+
row_type: Type[GenericNewPyArrowTable],
232232
operation: str,
233233
seq_of_parameters: "Sequence[Any]",
234234
/,
235235
**options: Unpack[ExecuteOptions],
236-
) -> "AsyncCursor[Never, GenericNewPandasDataFrame, Never]":
236+
) -> "AsyncCursor[Never, Never, GenericNewPyArrowTable]":
237237
...
238238

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

250250
@override
251251
async def executemany_map(
252252
self,
253253
row_type: Union[
254254
Type[GenericNewRowType],
255-
Type[GenericNewPanderaDataFrameModel],
256255
Type[GenericNewPandasDataFrame],
257256
Type[GenericNewPyArrowTable],
257+
Type[GenericNewPanderaDataFrameModel],
258258
],
259259
operation: str,
260260
seq_of_parameters: "Sequence[Any]",

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

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -176,44 +176,44 @@ def execute_map(
176176
@overload
177177
def execute_map(
178178
self,
179-
row_type: Type[GenericNewPanderaDataFrameModel],
179+
row_type: Type[GenericNewPandasDataFrame],
180180
operation: str,
181181
parameters: "Optional[Any]" = None,
182182
/,
183183
**options: Unpack[ExecuteOptions],
184-
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
184+
) -> Cursor[Never, GenericNewPandasDataFrame, Never]:
185185
...
186186

187187
@overload
188188
def execute_map(
189189
self,
190-
row_type: Type[GenericNewPandasDataFrame],
190+
row_type: Type[GenericNewPyArrowTable],
191191
operation: str,
192192
parameters: "Optional[Any]" = None,
193193
/,
194194
**options: Unpack[ExecuteOptions],
195-
) -> Cursor[Never, GenericNewPandasDataFrame, Never]:
195+
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
196196
...
197197

198198
@overload
199199
def execute_map(
200200
self,
201-
row_type: Type[GenericNewPyArrowTable],
201+
row_type: Type[GenericNewPanderaDataFrameModel],
202202
operation: str,
203203
parameters: "Optional[Any]" = None,
204204
/,
205205
**options: Unpack[ExecuteOptions],
206-
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
206+
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
207207
...
208208

209209
@override
210210
def execute_map(
211211
self,
212212
row_type: Union[
213213
Type[GenericNewRowType],
214-
Type[GenericNewPanderaDataFrameModel],
215214
Type[GenericNewPandasDataFrame],
216215
Type[GenericNewPyArrowTable],
216+
Type[GenericNewPanderaDataFrameModel],
217217
],
218218
operation: str,
219219
parameters: "Optional[Any]" = None,
@@ -260,44 +260,44 @@ def executemany_map(
260260
@overload
261261
def executemany_map(
262262
self,
263-
row_type: Type[GenericNewPanderaDataFrameModel],
263+
row_type: Type[GenericNewPandasDataFrame],
264264
operation: str,
265265
seq_of_parameters: Sequence[Any],
266266
/,
267267
**options: Unpack[ExecuteOptions],
268-
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
268+
) -> Cursor[Never, GenericNewPandasDataFrame, Never]:
269269
...
270270

271271
@overload
272272
def executemany_map(
273273
self,
274-
row_type: Type[GenericNewPandasDataFrame],
274+
row_type: Type[GenericNewPyArrowTable],
275275
operation: str,
276276
seq_of_parameters: Sequence[Any],
277277
/,
278278
**options: Unpack[ExecuteOptions],
279-
) -> Cursor[Never, GenericNewPandasDataFrame, Never]:
279+
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
280280
...
281281

282282
@overload
283283
def executemany_map(
284284
self,
285-
row_type: Type[GenericNewPyArrowTable],
285+
row_type: Type[GenericNewPanderaDataFrameModel],
286286
operation: str,
287287
seq_of_parameters: Sequence[Any],
288288
/,
289289
**options: Unpack[ExecuteOptions],
290-
) -> Cursor[Never, Never, GenericNewPyArrowTable]:
290+
) -> Cursor[Never, PanderaDataFrame[GenericNewPanderaDataFrameModel], Never]:
291291
...
292292

293293
@override
294294
def executemany_map(
295295
self,
296296
row_type: Union[
297297
Type[GenericNewRowType],
298-
Type[GenericNewPanderaDataFrameModel],
299298
Type[GenericNewPandasDataFrame],
300299
Type[GenericNewPyArrowTable],
300+
Type[GenericNewPanderaDataFrameModel],
301301
],
302302
operation: str,
303303
seq_of_parameters: Sequence[Any],

0 commit comments

Comments
 (0)