Improve type hints for UserSerializer (#4429)

* Improve type hints for UserSerializer * Fix NameError for UserType
cookiecutter · Jul 11, 2023 · 67a7194 · 67a7194
1 parent a78f273
commit 67a7194
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/api/serializers.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/api/serializers.py
@@ -1,10 +1,13 @@
 from django.contrib.auth import get_user_model
 from rest_framework import serializers
 
+from {{ cookiecutter.project_slug }}.users.models import User as UserType
+
+
 User = get_user_model()
 
 
-class UserSerializer(serializers.ModelSerializer):
+class UserSerializer(serializers.ModelSerializer[UserType]):
     class Meta:
         model = User
         {%- if cookiecutter.username_type == "email" %}