Skip to content

Commit 1b741bb

Browse files
committed
Fixed test typing
1 parent c6c842f commit 1b741bb

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

tests/panels/sqlalchemy/crud.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from sqlalchemy import Column
12
from sqlalchemy.orm import Session
23

34
from . import models
@@ -10,5 +11,9 @@ def create_user(db: Session, username: str) -> models.User:
1011
return user
1112

1213

13-
def get_user(db: Session, user_id: int) -> models.User:
14-
return db.query(models.User).filter(models.User.id == user_id).first()
14+
def get_user(db: Session, user_id: Column[int]) -> models.User:
15+
query = db.query(models.User).filter(models.User.id == user_id)
16+
17+
if query is not None:
18+
return query.first()
19+
return query

0 commit comments

Comments
 (0)