@@ -29,8 +29,8 @@ def __init__(self, data: Any) -> None:
29
29
@overload
30
30
def write_json (self , file : str ) -> None : ...
31
31
@overload
32
- def write_json (self , file : None ) -> str : ...
33
- def write_json (self , file = None ):
32
+ def write_json (self , file : None = ... ) -> str : ...
33
+ def write_json (self , file : str | None = None ) -> str | None :
34
34
if file is None :
35
35
msg = (
36
36
f"Writing to JSON string rather than file is not supported for "
@@ -139,7 +139,7 @@ def head(self, n: int) -> Self:
139
139
def write_json (self , file : str ) -> None : ...
140
140
@overload
141
141
def write_json (self , file : None ) -> str : ...
142
- def write_json (self , file = None ):
142
+ def write_json (self , file : str | None = None ) -> str | None :
143
143
if file is not None :
144
144
msg = (
145
145
f"Writing to file rather than JSON string is not supported for "
@@ -159,13 +159,11 @@ def write_feather(self, file: str) -> None:
159
159
self ._d .to_feather (file )
160
160
161
161
162
- @overload
163
- def _create_adaptor (obj : Any ) -> _Adaptor : ...
164
162
@overload
165
163
def _create_adaptor (obj : _DataFrame ) -> _DFAdaptor : ...
166
164
@overload
167
- def _create_adaptor (obj : _PandasDataFrame ) -> _PandasAdaptor : ...
168
- def _create_adaptor (obj ) :
165
+ def _create_adaptor (obj : Any ) -> _Adaptor : ...
166
+ def _create_adaptor (obj : Any | _DataFrame ) -> _Adaptor | _DFAdaptor :
169
167
if isinstance (obj , _AbstractPandasFrame ):
170
168
return _PandasAdaptor (obj )
171
169
else :
0 commit comments