Skip to content

Commit

Permalink
Adding additional columns to help support mlos_benchd_service - See m…
Browse files Browse the repository at this point in the history
  • Loading branch information
bpkroth committed Jan 14, 2025
1 parent 650862e commit bdb5de1
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions mlos_bench/mlos_bench/storage/sql/schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,6 +103,16 @@ def __init__(self, engine: Engine | None):
Column("root_env_config", String(1024), nullable=False),
Column("git_repo", String(1024), nullable=False),
Column("git_commit", String(40), nullable=False),
Column("ts_start", DateTime, nullable=False),
Column("ts_end", DateTime),
# Should match the text IDs of `mlos_bench.environments.Status` enum:
Column("status", String(self._STATUS_LEN), nullable=False),
# There may be more than one mlos_benchd_service running on different hosts.
# This column stores the hostname of the worker that picked up the experiment.
# They should use a transaction to update it to their own hostname when
# they start if and only if its NULL.
Column("experiment_worker_name", String(40), nullable=True, comment="Worker Hostname"),
Column("experiment_worker_id", Integer, nullable=True, comment="Worker Process ID"),
PrimaryKeyConstraint("exp_id"),
)
"""The Table storing
Expand Down

0 comments on commit bdb5de1

Please sign in to comment.