fixed errors with imports

deobuxay · Mar 1, 2017 · cc2dc41 · cc2dc41
1 parent 2ce0318
commit cc2dc41
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rest_framework_jwt/utils.py b/rest_framework_jwt/utils.py
@@ -4,7 +4,11 @@
 
 from six import string_types
 
-from django.db.models.loading import get_model
+try:
+    from django.db.models.loading import get_model
+except ImportError:
+    from django.apps import apps
+    get_model = apps.get_model
 
 from calendar import timegm
 from datetime import datetime