Skip to content

Commit 5258da9

Browse files
Update type hinting and update doc
1 parent cc0fe3b commit 5258da9

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

pandas/io/sql.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from datetime import date, datetime, time
88
from functools import partial
99
import re
10-
from typing import Iterator, Optional, Union, overload
10+
from typing import Dict, Iterator, Optional, Union, overload
1111
import warnings
1212

1313
import numpy as np
@@ -301,7 +301,7 @@ def read_sql_query(
301301
params=None,
302302
parse_dates=None,
303303
chunksize: None = None,
304-
dtype: Optional[Dtype] = None,
304+
dtype: Optional[Union[Dtype, Dict[str, Dtype]]] = None,
305305
) -> DataFrame:
306306
...
307307

@@ -315,7 +315,7 @@ def read_sql_query(
315315
params=None,
316316
parse_dates=None,
317317
chunksize: int = 1,
318-
dtype: Optional[Dtype] = None,
318+
dtype: Optional[Union[Dtype, Dict[str, Dtype]]] = None,
319319
) -> Iterator[DataFrame]:
320320
...
321321

@@ -328,7 +328,7 @@ def read_sql_query(
328328
params=None,
329329
parse_dates=None,
330330
chunksize: Optional[int] = None,
331-
dtype: Optional[Dtype] = None,
331+
dtype: Optional[Union[Dtype, Dict[str, Dtype]]] = None,
332332
) -> Union[DataFrame, Iterator[DataFrame]]:
333333
"""
334334
Read SQL query into a DataFrame.
@@ -367,6 +367,9 @@ def read_sql_query(
367367
chunksize : int, default None
368368
If specified, return an iterator where `chunksize` is the number of
369369
rows to include in each chunk.
370+
dtype : Type name or dict of columns
371+
Data type for data or columns. E.g. np.float64 or
372+
{‘a’: np.float64, ‘b’: np.int32, ‘c’: ‘Int64’}
370373
371374
Returns
372375
-------

0 commit comments

Comments
 (0)