Resolve test_multi_pk.py Flask-SQLAlchemy Model DeprecationWarning

pallets-eco · Oct 18, 2023 · d9a1db6 · d9a1db6
1 parent 5212734
commit d9a1db6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/flask_admin/tests/sqla/test_multi_pk.py b/flask_admin/tests/sqla/test_multi_pk.py
@@ -1,7 +1,11 @@
 from . import setup
 from .test_basic import CustomModelView
 
-from flask_sqlalchemy import Model
+try:
+    # Flask-SQLAlchemy 3
+    from flask_sqlalchemy.model import Model
+except ImportError:
+    from flask_sqlalchemy import Model
 from sqlalchemy.orm import declarative_base