Bump Version 0.5.9

superduper-io · Jan 30, 2025 · b6a9820 · b6a9820
1 parent d60f4a8
commit b6a9820
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 name = "superduper-framework"
 description = "🔮 Bring AI to your favourite database 🔮"
 readme = "README.md"
-version = '0.5.8'
+version = '0.5.9'
 license = {file = "LICENSE"}
 maintainers = [{name = "superduper.io, Inc.", email = "[email protected]"}]
 keywords = [

diff --git a/superduper/components/model.py b/superduper/components/model.py
@@ -573,7 +573,11 @@ def _prepare_inputs_from_select(
         X_data: t.Any
         mapping = Mapping(X, self.signature)
         if in_memory:
+            logging.info('Loading data into memory...')
             docs = list(self.db.execute(select.select_using_ids(ids)))
+            pid = select.table_or_collection.primary_id
+            lookup = {r[pid]: r for r in docs}
+            docs = [lookup[id_] for id_ in ids]
             X_data = list(map(lambda x: mapping(x), docs))
         else:
             assert isinstance(self.db, Datalayer)