...

tests/test_enviformer.py

@@ -1,10 +1,15 @@
 from collections import defaultdict
 from datetime import datetime
 from tempfile import TemporaryDirectory
+
+from django.conf import settings as s
 from django.test import TestCase, tag
+
 from epdb.logic import PackageManager
-from epdb.models import User, EnviFormer, Package, Setting
-from epdb.tasks import predict_simple, predict
+from epdb.models import EnviFormer, Setting, User
+from epdb.tasks import predict, predict_simple
+
+Package = s.GET_PACKAGE_MODEL()
 
 
 def measure_predict(mod, pathway_pk=None):