Use bakery if present

Closes alsur#3
jayvdb · Jun 15, 2020 · 04c71ed · 04c71ed
1 parent 8baf77d
commit 04c71ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/admin_auto_tests/test_model.py b/admin_auto_tests/test_model.py
@@ -2,7 +2,10 @@
 
 import factory
 from django.contrib.auth import get_user_model
-from model_mommy import mommy
+try:
+    from model_bakery import bakery as mommy
+except ImportError:
+    from model_mommy import mommy
 
 try:
     from django.core.urlresolvers import reverse