fix: Move objproxies import in test branch (#7058)

fossasia · Jun 11, 2020 · 8f62b87 · 8f62b87
1 parent ab4d152
commit 8f62b87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/models/__init__.py b/app/models/__init__.py
@@ -1,7 +1,6 @@
 import sys
 
 from flask_sqlalchemy import SQLAlchemy
-from objproxies import CallbackProxy
 from sqlalchemy_continuum import make_versioned
 from sqlalchemy_continuum.plugins import FlaskPlugin
 
@@ -13,5 +12,6 @@
 # to a nested transaction which is rolled backed
 # after each test
 if 'pytest' in sys.modules:
+    from objproxies import CallbackProxy
     db._session = db.session
     db.session = CallbackProxy(lambda: db._session)