fixup

microsoft · Jan 6, 2025 · 23ea0de · 23ea0de
1 parent 94b327f
commit 23ea0de
Showing 1 changed file with 1 addition and 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
@@ -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)