Merge pull request #2491 from cjmayo/test_multi_pk_2

Revert "Add SQLAlchemy 1 support back to test_multi_pk.py"
pallets-eco · Jul 30, 2024 · aaaa32e · aaaa32e
2 parents 1cd3621 + 9ea52a8
commit aaaa32e
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/flask_admin/tests/sqla/test_multi_pk.py b/flask_admin/tests/sqla/test_multi_pk.py
@@ -1,11 +1,7 @@
 from .test_basic import CustomModelView
 
 from flask_sqlalchemy.model import Model
-try:
-    # SQLAlchemy 2.0
-    from sqlalchemy.orm import declarative_base
-except ImportError:
-    from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import declarative_base
 
 
 def test_multiple_pk(app, db, admin):