corvic-engine 0.3.0rc73__tar.gz → 0.3.0rc75__tar.gz

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/Cargo.lock

@@ -16,7 +16,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.73"
+version = "0.3.0-rc.75"
 dependencies = [
  "numpy",
  "pyo3",
@@ -57,7 +57,7 @@ checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.73"
+version = "0.3.0-rc.75"
 dependencies = [
  "version_check",
 ]

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/Cargo.toml

@@ -3,7 +3,7 @@ resolver = "2"
 members = ["python"]
 
 [workspace.package]
-version = "0.3.0-rc.73"
+version = "0.3.0-rc.75"
 edition = "2021"
 publish = false
 

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corvic-engine
-Version: 0.3.0rc73
+Version: 0.3.0rc75
 Classifier: Environment :: Console
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: Implementation :: CPython

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/pyproject.toml

@@ -18,7 +18,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.73" # sync this with tool.poetry.version below
+version = "0.3.0-rc.75" # sync this with tool.poetry.version below
 
 # TODO(ddn): Pull dependencies from tool.poetry.dependencies. We use maturin as
 # the build system but poetry for the development environment.
@@ -72,7 +72,7 @@ packages = [
 ]
 description = ""
 authors = []
-version = "0.3.0-rc.73"
+version = "0.3.0-rc.75"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/python/corvic/model/_proto_orm_convert.py

@@ -5,7 +5,6 @@ from typing import Any, Protocol, TypeVar
 import sqlalchemy as sa
 import sqlalchemy.orm as sa_orm
 from google.protobuf import timestamp_pb2
-from more_itertools import flatten
 
 from corvic import orm
 from corvic.result import InternalError, InvalidArgumentError, Ok
@@ -345,7 +344,6 @@ def _ensure_id(
 def pipeline_proto_to_orm(  # noqa: C901
     proto_obj: models_pb2.Pipeline, session: sa_orm.Session
 ) -> Ok[orm.Pipeline] | orm.InvalidORMIdentifierError | InvalidArgumentError:
-    inputs = list[orm.PipelineInput]()
     orm_obj = orm.Pipeline(
         name=proto_obj.name,
         transformation=proto_obj.pipeline_transformation,
@@ -381,9 +379,9 @@ def pipeline_proto_to_orm(  # noqa: C901
 
     if proto_obj.org_id:
         org_id = orm.OrgID(proto_obj.org_id)
-        for obj in flatten((inputs, outputs)):
+        for obj in outputs:
             obj.org_id = org_id
-    for obj in flatten((inputs, outputs)):
+    for obj in outputs:
         session.merge(obj)
     return Ok(orm_obj)
 

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/python/corvic/orm/__init__.py

@@ -89,6 +89,7 @@ class BelongsToRoomMixin(sa_orm.MappedAsDataclass):
         ForeignKey(Room).make(ondelete="CASCADE"),
         nullable=True,
         default=None,
+        index=True,
     )
 
 

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/python/corvic/orm/base.py

@@ -134,7 +134,11 @@ class Base(sa_orm.MappedAsDataclass, sa_orm.DeclarativeBase):
     }
 
     _created_at: sa_orm.Mapped[datetime] = sa_orm.mapped_column(
-        "created_at", sa.DateTime(timezone=True), server_default=utc_now(), init=False
+        "created_at",
+        sa.DateTime(timezone=True),
+        server_default=utc_now(),
+        init=False,
+        index=True,
     )
 
     _updated_at: sa_orm.Mapped[datetime] = sa_orm.mapped_column(

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/python/corvic/orm/mixins.py

@@ -229,6 +229,7 @@ class BelongsToOrgMixin(sa_orm.MappedAsDataclass):
         nullable=False,
         default_factory=_make_org_id_default,
         init=False,
+        index=True,
     )
 
     @sa_orm.validates("org_id")

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/python/corvic/system/_dimension_reduction.py

@@ -1,3 +1,4 @@
+import sys
 from typing import Any, Protocol
 
 import numpy as np
@@ -71,6 +72,7 @@ class UmapDimensionReducer(DimensionReducer):
             init=init,
             low_memory=False,
             verbose=False,
+            tqdm_kwds={"file": sys.stdout},
         )
         return Ok(projector.fit_transform(vectors))
 

{corvic_engine-0.3.0rc73 → corvic_engine-0.3.0rc75}/python/corvic/system/in_memory_executor.py

@@ -775,8 +775,9 @@ class InMemoryExecutor(OpGraphExecutor):
         right_lf = right_lf.select(
             [
                 col
-                for col in right_lf.columns
-                if col in op.right_join_columns or col not in left_lf.columns
+                for col in right_lf.collect_schema().names()
+                if col in op.right_join_columns
+                or col not in left_lf.collect_schema().names()
             ]
         )
         metrics = right_lfm.metrics.copy()