Hardcode the use of auth.Models over get_user_model. (Closes: #1)

lamby · Sep 12, 2018 · f06d23f · f06d23f
1 parent 7a93da7
commit f06d23f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/django_auto_one_to_one/models.py b/django_auto_one_to_one/models.py
@@ -1,8 +1,8 @@
 import six
 
 from django.db import models
-from django.contrib.auth import get_user_model
 from django.db.models.signals import post_save, pre_delete
+from django.contrib.auth.models import User
 
 
 def AutoOneToOneModel(parent, related_name=None, attr=None, on_delete=models.CASCADE, auto=True):
@@ -136,4 +136,4 @@ def __str__(self):
 
 
 def PerUserData(*args, **kwargs):
-    return AutoOneToOneModel(get_user_model(), *args, **kwargs)
+    return AutoOneToOneModel(User, *args, **kwargs)