mlos_bench.storage.sql.common module

Common SQL methods for accessing the stored benchmark data.

mlos_bench.storage.sql.common.get_results_df(engine: Engine, schema: DbSchema, experiment_id: str, tunable_config_id: int | None = None) → DataFrame

Gets TrialData for the given experiment_data and optionally additionally restricted by tunable_config_id.

Used by both TunableConfigTrialGroupSqlData and ExperimentSqlData.

mlos_bench.storage.sql.common.get_trials(engine: Engine, schema: DbSchema, experiment_id: str, tunable_config_id: int | None = None) → Dict[int, TrialData]

Gets TrialData for the given experiment_data and optionally additionally restricted by tunable_config_id.

Used by both TunableConfigTrialGroupSqlData and ExperimentSqlData.