POST/GET /flows/{name}/nodes/{id}/train submit wren_train on the node's
targets, examples and tests and read its latest run (run.for_node,
cause "train"); "Use this model" writes `model` into the draft.
34 lines
679 B
Python
34 lines
679 B
Python
"""run.for_node
|
|
|
|
A wren node's Train button submits wren's training flow as a run. The node
|
|
it was pressed on is what ties that run back to the node's panel.
|
|
|
|
Revision ID: a3c7e5d91f42
|
|
Revises: ccb21eacf5d5
|
|
Create Date: 2026-10-24
|
|
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
import sqlmodel.sql.sqltypes
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "a3c7e5d91f42"
|
|
down_revision = "ccb21eacf5d5"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.add_column(
|
|
"run",
|
|
sa.Column(
|
|
"for_node", sqlmodel.sql.sqltypes.AutoString(length=255), nullable=True
|
|
),
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_column("run", "for_node")
|