Skip to content

Commit aa9749a

Browse files
author
Chris Bertinato
committed
Added annotations and benchmarks
1 parent d7879af commit aa9749a

File tree

2 files changed

+24
-17
lines changed

2 files changed

+24
-17
lines changed

asv_bench/benchmarks/io/json.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ class ToJSON(BaseIO):
6666
fname = "__test__.json"
6767
params = [
6868
["split", "columns", "index", "values", "records"],
69-
["df", "df_date_idx", "df_td_int_ts", "df_int_floats", "df_int_float_str"],
69+
["df", "df_date_idx", "df_td", "df_td_int_ts", "df_int_floats", "df_int_float_str"],
7070
]
7171
param_names = ["orient", "frame"]
7272

@@ -81,6 +81,13 @@ def setup(self, orient, frame):
8181
strings = tm.makeStringIndex(N)
8282
self.df = DataFrame(np.random.randn(N, ncols), index=np.arange(N))
8383
self.df_date_idx = DataFrame(np.random.randn(N, ncols), index=index)
84+
self.df_td = DataFrame(
85+
{
86+
"td_1": timedeltas,
87+
"td_2": timedeltas
88+
},
89+
index=index,
90+
)
8491
self.df_td_int_ts = DataFrame(
8592
{
8693
"td_1": timedeltas,

pandas/io/json/_json.py

+16-16
Original file line numberDiff line numberDiff line change
@@ -174,14 +174,14 @@ class SeriesWriter(Writer):
174174
def __init__(
175175
self,
176176
obj,
177-
orient,
178-
date_format,
179-
double_precision,
180-
ensure_ascii,
181-
date_unit,
182-
index,
183-
default_handler=None,
184-
indent=0,
177+
orient: Optional[str],
178+
date_format: str,
179+
double_precision: int,
180+
ensure_ascii: bool,
181+
date_unit: str,
182+
index: bool,
183+
default_handler: Optional[Callable[[Any], Serializable]] = None,
184+
indent: int = 0,
185185
):
186186
super().__init__(
187187
obj,
@@ -237,14 +237,14 @@ class FrameWriter(Writer):
237237
def __init__(
238238
self,
239239
obj,
240-
orient,
241-
date_format,
242-
double_precision,
243-
ensure_ascii,
244-
date_unit,
245-
index,
246-
default_handler=None,
247-
indent=0,
240+
orient: Optional[str],
241+
date_format: str,
242+
double_precision: int,
243+
ensure_ascii: bool,
244+
date_unit: str,
245+
index: bool,
246+
default_handler: Optional[Callable[[Any], Serializable]] = None,
247+
indent: int = 0,
248248
):
249249
super().__init__(
250250
obj,

0 commit comments

Comments
 (0)