From e28059f92ede617c6bc93eead6e019b995cd79fc Mon Sep 17 00:00:00 2001
From: huangsong <huangsong@leyantech.com>
Date: Fri, 14 Jan 2022 14:29:02 +0800
Subject: [PATCH] fix type annotation

---
 ormar/queryset/queryset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ormar/queryset/queryset.py b/ormar/queryset/queryset.py
index a1b4ce1..d5da089 100644
--- a/ormar/queryset/queryset.py
+++ b/ormar/queryset/queryset.py
@@ -1059,7 +1059,7 @@ class QuerySet(Generic[T]):
         await self.database.execute(expr)
 
         # FIXME: add pre_save signals
-        async def after_create(entity):
+        async def after_create(entity: "T") -> None:
             entity.set_save_status(True)
             await entity.signals.post_save.send(
                 sender=entity.__class__, instance=entity)
@@ -1133,7 +1133,7 @@ class QuerySet(Generic[T]):
         await self.database.execute_many(expr, ready_objects)
 
         # FIXME: add pre-update-signals
-        async def after_update(entity):
+        async def after_update(entity: "T") -> None:
             entity.set_save_status(True)
             await entity.signals.post_update.send(
                 sender=entity.__class__, instance=entity