From 3b7de8101b4ead4912ccdaaecd20da9c81ab6536 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fl=C3=A1vio=20Juvenal?= <flavio@vinta.com.br>
Date: Thu, 28 Sep 2023 14:49:27 -0300
Subject: [PATCH] Fix typing for test model

---
 tests/projects/models.py            | 2 +-
 tests/projects/snapshots/schema.gql | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/projects/models.py b/tests/projects/models.py
index 38115a8e..0c564aa7 100644
--- a/tests/projects/models.py
+++ b/tests/projects/models.py
@@ -51,7 +51,7 @@ class Status(models.TextChoices):
     )
 
     @model_property(annotate={"_milestone_count": Count("milestone")})
-    def is_small(self) -> int:
+    def is_small(self) -> bool:
         return self._milestone_count < 3
 
 
diff --git a/tests/projects/snapshots/schema.gql b/tests/projects/snapshots/schema.gql
index 7ebf8f89..989a303f 100644
--- a/tests/projects/snapshots/schema.gql
+++ b/tests/projects/snapshots/schema.gql
@@ -452,7 +452,7 @@ type ProjectType implements Node {
   milestonesCount: Int!
   isDelayed: Boolean!
   cost: Decimal @isAuthenticated
-  isSmall: Int!
+  isSmall: Boolean!
 }
 
 """An edge in a connection."""