Skip to content

Commit bcab904

Browse files
authored
add missing types to Session and Query (#233)
1 parent 81e44ae commit bcab904

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

sqlalchemy-stubs/orm/query.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ class Query(Generic[_T]):
4545
def with_entities(self, *entities): ...
4646
def add_columns(self, *column): ...
4747
def add_column(self, column): ...
48-
def options(self, *args): ...
48+
def options(self: _Q, *args) -> _Q: ...
4949
def with_transformation(self, fn): ...
5050
def with_hint(self, selectable, text, dialect_name: str = ...): ...
5151
def with_statement_hint(self, text, dialect_name: str = ...): ...

sqlalchemy-stubs/orm/session.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ class Session(_SessionClassMethods):
6161
mapper: Optional[Any] = ..., bind: Optional[Any] = ..., **kw): ...
6262
def close(self) -> None: ...
6363
def invalidate(self) -> None: ...
64-
def expunge_all(self): ...
64+
def expunge_all(self) -> None: ...
6565
def bind_mapper(self, mapper, bind): ...
6666
def bind_table(self, table, bind): ...
6767
def get_bind(self, mapper: Optional[Any] = ..., clause: Optional[Any] = ...): ...
@@ -77,7 +77,7 @@ class Session(_SessionClassMethods):
7777
def expire_all(self) -> None: ...
7878
def expire(self, instance, attribute_names: Optional[Any] = ...) -> None: ...
7979
def prune(self): ...
80-
def expunge(self, instance): ...
80+
def expunge(self, instance) -> None: ...
8181
def add(self, instance, _warn: bool = ...) -> None: ...
8282
def add_all(self, instances) -> None: ...
8383
def delete(self, instance): ...

0 commit comments

Comments
 (0)