From 23ea0dee5f0678ef914e908d7156758608f5654a Mon Sep 17 00:00:00 2001 From: Brian Kroth Date: Mon, 6 Jan 2025 23:34:25 +0000 Subject: [PATCH] fixup --- mlos_bench/mlos_bench/tests/storage/trial_data_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlos_bench/mlos_bench/tests/storage/trial_data_test.py b/mlos_bench/mlos_bench/tests/storage/trial_data_test.py index 77bc1eb2432..62c7bfb1bfc 100644 --- a/mlos_bench/mlos_bench/tests/storage/trial_data_test.py +++ b/mlos_bench/mlos_bench/tests/storage/trial_data_test.py @@ -21,7 +21,7 @@ def test_exp_trial_data(exp_data: ExperimentData) -> None: assert trial.tunable_config_id == expected_config_id assert trial.status == Status.SUCCEEDED assert trial.metadata_dict["repeat_i"] == 1 - assert trial.metadata_dict["trial_runner_id"] == "1" + assert trial.trial_runner_id == 1 assert list(trial.results_dict.keys()) == ["score"] assert trial.results_dict["score"] == pytest.approx(73.27, 0.01) assert isinstance(trial.ts_start, datetime)