fix exception

2025-05-24 11:29:37 -03:00
parent c7cf30859a
commit be177a9b40
3 changed files with 4 additions and 4 deletions
--- a/layercake/layercake/dynamodb.py
+++ b/layercake/layercake/dynamodb.py
@@ -313,8 +313,8 @@ class TransactKey:
    pairs: tuple[KeyPair, ...] = ()

    def __add__(self, other: SortKey | KeyPair) -> 'TransactKey':
-        if not isinstance(other, SortKey):
-            raise TypeError('Can only add a SortKey to a TransactKey')
+        if not isinstance(other, (SortKey, KeyPair)):
+            raise TypeError('Can only add a SortKey and KeyPair to a TransactKey')

        if isinstance(other, SortKey):
            other = KeyPair(self.pk, other)