rebase

dgilland · Oct 3, 2023 · 370bc41 · 370bc41
1 parent 5fdcb65
commit 370bc41
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/fixtures.py b/tests/fixtures.py
@@ -8,11 +8,18 @@
 import pytest_asyncio
 import sqlalchemy as sa
 from sqlalchemy.orm import RelationshipProperty
-from sqlalchemy.orm.collections import attribute_keyed_dict
 
 from sqlservice import AsyncDatabase, Database, ModelBase, as_declarative
 
 
+try:
+    from sqlalchemy.orm.collections import attribute_keyed_dict
+except ImportError:
+    from sqlalchemy.orm.collections import (  # type: ignore
+        attribute_mapped_collection as attribute_keyed_dict,
+    )
+
+
 def unique_id():
     if not hasattr(unique_id, "_id"):
         unique_id._id = 0