temporal_v2: add memory cost reporting (#4257)

Signed-off-by: Benji Visser <benji@093b.org>
Co-authored-by: Benji Visser <benji@093b.org>
This commit is contained in:
Shuchang Zheng
2025-12-10 11:30:41 -08:00
committed by GitHub
parent c281428d73
commit e8251701cd
3 changed files with 35 additions and 0 deletions

View File

@@ -0,0 +1,31 @@
"""add memory_mb to task_runs
Revision ID: 174dcd456325
Revises: 67784da0203e
Create Date: 2025-12-10 19:15:24.067819+00:00
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "174dcd456325"
down_revision: Union[str, None] = "67784da0203e"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.add_column("task_runs", sa.Column("memory_mb", sa.Integer(), nullable=True))
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column("task_runs", "memory_mb")
# ### end Alembic commands ###