corvic-engine 0.3.0rc66__tar.gz → 0.3.0rc67__tar.gz

{corvic_engine-0.3.0rc66 → corvic_engine-0.3.0rc67}/Cargo.lock

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

{corvic_engine-0.3.0rc66 → corvic_engine-0.3.0rc67}/Cargo.toml

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

{corvic_engine-0.3.0rc66 → corvic_engine-0.3.0rc67}/PKG-INFO

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

{corvic_engine-0.3.0rc66 → corvic_engine-0.3.0rc67}/pyproject.toml

@@ -20,7 +20,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.66" # sync this with tool.poetry.version below
+version = "0.3.0-rc.67" # 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.
@@ -74,7 +74,7 @@ packages = [
 ]
 description = ""
 authors = []
-version = "0.3.0-rc.66"
+version = "0.3.0-rc.67"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"

{corvic_engine-0.3.0rc66 → corvic_engine-0.3.0rc67}/python/corvic/orm/mixins.py

@@ -10,6 +10,7 @@ import sqlalchemy as sa
 from google.protobuf import timestamp_pb2
 from sqlalchemy import event, exc
 from sqlalchemy import orm as sa_orm
+from sqlalchemy.ext import hybrid
 from sqlalchemy.ext.hybrid import hybrid_property
 from typing_extensions import LiteralString
 
@@ -121,7 +122,8 @@ class SoftDeleteMixin(sa_orm.MappedAsDataclass):
     class ForceLoadOption(sa_orm.UserDefinedOption):
         """Option for ignoring soft delete status when loading."""
 
-    deleted_at: sa_orm.Mapped[datetime | None] = sa_orm.mapped_column(
+    _deleted_at: sa_orm.Mapped[datetime | None] = sa_orm.mapped_column(
+        "deleted_at",
         sa.DateTime(timezone=True),
         server_default=None,
         default=None,
@@ -131,6 +133,15 @@ class SoftDeleteMixin(sa_orm.MappedAsDataclass):
         default=True,
     )
 
+    @hybrid.hybrid_property
+    def deleted_at(self) -> datetime | None:
+        if not self._deleted_at:
+            return None
+        return self._deleted_at.replace(tzinfo=timezone.utc)
+
+    def reset_delete(self):
+        self._deleted_at = None
+
     @classmethod
     def _force_load_option(cls):
         return cls.ForceLoadOption()
@@ -159,7 +170,7 @@ class SoftDeleteMixin(sa_orm.MappedAsDataclass):
         # set is_live to None instead of False so that orm objects can use it to
         # build uniqueness constraints that are only enforced on non-deleted objects
         self.is_live = None
-        self.deleted_at = datetime.now(tz=timezone.utc)
+        self._deleted_at = datetime.now(tz=timezone.utc)
 
     @hybrid_property
     def is_deleted(self) -> bool:
@@ -296,7 +307,7 @@ class Session(sa_orm.Session):
         if self._soft_deleted:
             for obj in self._soft_deleted.values():
                 if isinstance(obj, SoftDeleteMixin):
-                    obj.deleted_at = None
+                    obj.reset_delete()
                     obj.is_live = True
                     continue
                 raise RuntimeError("non-soft delete object in soft deleted set")

{corvic_engine-0.3.0rc66 → corvic_engine-0.3.0rc67}/python/corvic/sql/parse_ops.py

@@ -51,7 +51,11 @@ class NoRowsError(FailedPreconditionError):
 
 def sources_are_empty(op: op_graph.Op) -> bool:
     match op:
-        case op_graph.op.SelectFromStaging() | op_graph.op.ReadFromParquet():
+        case (
+            op_graph.op.SelectFromStaging()
+            | op_graph.op.ReadFromParquet()
+            | op_graph.op.SelectFromVectorStaging()
+        ):
             return False
         case op_graph.op.Empty():
             return True