[Model Registry] Implement ModelVersion stage state machine

astroml/db/schema.py

@@ -537,24 +537,46 @@ class NormalizedTransaction(Base):
     )
 
 
+# ---------------------------------------------------------------------------
+# Ledger Processing
+# ---------------------------------------------------------------------------
+
 class ProcessedLedger(Base):
     """Tracking table for processed ledgers during backfill to ensure idempotency."""
 
     __tablename__ = "processed_ledgers"
 
     id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
     ledger_sequence: Mapped[int] = mapped_column(Integer, unique=True, nullable=False)
-    source: Mapped[str] = mapped_column(String(256), nullable=False, doc="Source of the ledger data (e.g., file path, API endpoint)")
-    processed_at: Mapped[datetime] = mapped_column(nullable=False, server_default=func.now())
-    status: Mapped[Literal["pending", "processing", "completed", "failed"]] = mapped_column(
-        String(32), nullable=False, server_default="pending"
+    source: Mapped[str] = mapped_column(
+        String(256),
+        nullable=False,
+        doc="Source of the ledger data (e.g., file path, API endpoint)",
+    )
+    processed_at: Mapped[datetime] = mapped_column(
+        nullable=False,
+        server_default=func.now(),
+    )
+    status: Mapped[
+        Literal["pending", "processing", "completed", "failed"]
+    ] = mapped_column(
+        String(32),
+        nullable=False,
+        server_default="pending",
     )
     error_message: Mapped[Optional[str]] = mapped_column(Text)
-    num_operations: Mapped[Optional[int]] = mapped_column(Integer, doc="Number of operations processed from this ledger")
-    num_transactions: Mapped[Optional[int]] = mapped_column(Integer, doc="Number of transactions processed from this ledger")
+    num_operations: Mapped[Optional[int]] = mapped_column(
+        Integer,
+        doc="Number of operations processed from this ledger",
+    )
+    num_transactions: Mapped[Optional[int]] = mapped_column(
+        Integer,
+        doc="Number of transactions processed from this ledger",
+    )
 
     __table_args__ = (
         Index("ix_processed_ledgers_ledger_sequence", "ledger_sequence"),
         Index("ix_processed_ledgers_status", "status"),
         Index("ix_processed_ledgers_source", "source"),
     )
+    )

astroml/tracking/__init__.py

@@ -1,3 +1,4 @@
 from .mlflow_tracker import MLflowTracker
+from .model_registry import ModelRegistry
 
-__all__ = ["MLflowTracker"]
+__all__ = ["MLflowTracker", "ModelRegistry"]