Registry

class stable_pretraining.registry.Registry(store: Store)

Bases: object

Read-only query interface over the registry cache.

Instantiate via open_registry() rather than directly — the factory runs a lazy filesystem scan first so you don’t query stale data.

query(*, tag: str | None = None, status: str | None = None, alive: bool | None = None, hparams: Dict[str, Any] | None = None, sort_by: str | None = None, descending: bool = True, limit: int | None = None) → List[RunRecord]

Query runs matching filters.

Parameters:

• tag – Include runs that carry this tag (uses substring match on the stored JSON tag array).

• status – Filter by status column (running, completed, failed, orphaned, interrupted).

• alive – Filter by heartbeat-based liveness.

• hparams – {key: value} pairs the flattened hparams must match (AND, client-side).

• sort_by – Column name or summary.<k> / hparams.<k> / config.<k>.

• descending – Sort order.

• limit – Max rows.

to_dataframe(**query_kwargs: Any)

Return a DataFrame with flattened hparams.* / summary.* cols.