Skip to content

Commit 5987e62

Browse files
Pass pyright
1 parent 58d427c commit 5987e62

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

pins/_adaptors.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ def __init__(self, data: Any) -> None:
2929
@overload
3030
def write_json(self, file: str) -> None: ...
3131
@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:
3434
if file is None:
3535
msg = (
3636
f"Writing to JSON string rather than file is not supported for "
@@ -139,7 +139,7 @@ def head(self, n: int) -> Self:
139139
def write_json(self, file: str) -> None: ...
140140
@overload
141141
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:
143143
if file is not None:
144144
msg = (
145145
f"Writing to file rather than JSON string is not supported for "
@@ -159,13 +159,11 @@ def write_feather(self, file: str) -> None:
159159
self._d.to_feather(file)
160160

161161

162-
@overload
163-
def _create_adaptor(obj: Any) -> _Adaptor: ...
164162
@overload
165163
def _create_adaptor(obj: _DataFrame) -> _DFAdaptor: ...
166164
@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:
169167
if isinstance(obj, _AbstractPandasFrame):
170168
return _PandasAdaptor(obj)
171169
else:

0 commit comments

Comments
 (0)