fixup

microsoft · Nov 25, 2024 · c193718 · c193718
1 parent 7a4f50e
commit c193718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlos_core/mlos_core/tests/optimizers/bayesian_optimizers_test.py b/mlos_core/mlos_core/tests/optimizers/bayesian_optimizers_test.py
@@ -39,8 +39,8 @@ def test_context_not_implemented_warning(
     suggestion = optimizer.suggest()
     scores = pd.Series({"score": [1]})
     context = pd.Series([["something"]])
-    suggestion.context = context
 
+    suggestion._context = context  # pylint: disable=protected-access
     with pytest.raises(UserWarning):
         optimizer.register(observations=suggestion.complete(scores))