a650097f87c8f442d7f81d3024b56f0e81910d06,mlflow/store/db_migrations/versions/cfd24bdc0731_update_run_status_constraint_with_killed.py,,,#,14

Before Change

    RunStatus.to_string(RunStatus.KILLED)
]

previous_run_status = [
    RunStatus.to_string(RunStatus.SCHEDULED),
    RunStatus.to_string(RunStatus.FAILED),
    RunStatus.to_string(RunStatus.FINISHED),
    RunStatus.to_string(RunStatus.RUNNING)
]


def _has_check_constraints():
    inspector = Inspector(context.get_bind().engine)

After Change

// the collection of CHECK constraints that should be preserved when performing the migration.
// The "status" constraint is excluded from this set because it is explicitly modified
// within the migration"s `upgrade()` routine.
check_constraint_table_args = [
    CheckConstraint(SqlRun.source_type.in_(SourceTypes), name="source_type"),
    CheckConstraint(SqlRun.lifecycle_stage.in_(LifecycleStage.view_type_to_stages(ViewType.ALL)),
                    name="runs_lifecycle_stage"),
]


def upgrade():
    with op.batch_alter_table("runs", table_args=check_constraint_table_args) as batch_op:

Instances