fix: type hints

ocadotechnology · Feb 8, 2024 · dd2705c · dd2705c
1 parent c5b304e
commit dd2705c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/codeforlife/user/models/teacher.py b/codeforlife/user/models/teacher.py
@@ -4,6 +4,7 @@
 """
 
 from common.models import Teacher, TeacherModelManager
+from django.db import models
 from django_stubs_ext.db.models import TypedModelMeta
 
 from .school import School
@@ -22,7 +23,7 @@ class Manager(TeacherModelManager):
         def get_queryset(self):
             return super().get_queryset().filter(school__isnull=False)
 
-    objects: Manager = Manager()
+    objects: models.Manager["SchoolTeacher"] = Manager()
 
 
 class NonSchoolTeacher(Teacher):
@@ -38,4 +39,4 @@ class Manager(TeacherModelManager):
         def get_queryset(self):
             return super().get_queryset().filter(school__isnull=True)
 
-    objects: Manager = Manager()
+    objects: models.Manager["NonSchoolTeacher"] = Manager()