Bump Version 0.5.8

superduper-io · Jan 30, 2025 · 4364455 · 4364455
1 parent 2dad709
commit 4364455
Show file tree

Hide file tree

Showing 2 changed files with 9 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.7'
+version = '0.5.8'
 license = {file = "LICENSE"}
 maintainers = [{name = "superduper.io, Inc.", email = "opensource@superduper.com"}]
 keywords = [

diff --git a/superduper/backends/query_dataset.py b/superduper/backends/query_dataset.py
@@ -64,6 +64,14 @@ def __init__(
                 self._documents = list(
                     self.db.execute(self.select.select_using_ids(ids))
                 )
+                primary_id = self.select.id_field
+                # is it possible that the _documents came out in a different order?
+                # if so, resort them
+                lookup = {
+                    r[primary_id]: r for r in self._documents
+                }
+                self._documents = [lookup[id] for id in ids]
+                # DONE
         else:
             if ids is None:
                 self._ids = [