fix: use base item type

experimaestro · Mar 5, 2024 · 605ed53 · 605ed53
1 parent e2e3c77
commit 605ed53
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/datamaestro/record.py b/src/datamaestro/record.py
@@ -228,7 +228,9 @@ def update(self, record: Record, *items: Item, cls=None):
                     "Updating unpickled records is not recommended"
                     " (speed issues): use the pickle record class as the cls input"
                 )
-                itemtypes = frozenset(record.items.keys())
+                itemtypes = frozenset(
+                    itemtype.__get_base__() for itemtype in record.items.keys()
+                )
                 cls = Record.fromitemtypes(itemtypes)
         else:
             assert (