diff --git a/codeforlife/tests/api.py b/codeforlife/tests/api.py index a99cbae..89c771b 100644 --- a/codeforlife/tests/api.py +++ b/codeforlife/tests/api.py @@ -268,6 +268,8 @@ def login_as(self, user: TypedUser, password: str = "password"): user: The user to log in as. password: The user's password. """ + auth_user = None + if isinstance(user, AdminSchoolTeacherUser): auth_user = self.login_admin_school_teacher(user.email, password) elif isinstance(user, NonAdminSchoolTeacherUser): diff --git a/codeforlife/user/models/teacher.py b/codeforlife/user/models/teacher.py index 0b816a8..3ce2d69 100644 --- a/codeforlife/user/models/teacher.py +++ b/codeforlife/user/models/teacher.py @@ -174,6 +174,7 @@ def get_queryset(self): objects: models.Manager["NonSchoolTeacher"] = Manager() +# pylint: disable-next=invalid-name TypedTeacher = t.Union[ SchoolTeacher, AdminSchoolTeacher, diff --git a/codeforlife/user/models/user.py b/codeforlife/user/models/user.py index 050d8e5..881d4bb 100644 --- a/codeforlife/user/models/user.py +++ b/codeforlife/user/models/user.py @@ -605,6 +605,7 @@ class Meta(TypedModelMeta): ) +# pylint: disable-next=invalid-name TypedUser = t.Union[ TeacherUser, SchoolTeacherUser,