flaml version fixups

microsoft · Jan 10, 2025 · a1274be · a1274be
1 parent 9c93cad
commit a1274be
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/mlos_bench/mlos_bench/tests/optimizers/toy_optimization_loop_test.py b/mlos_bench/mlos_bench/tests/optimizers/toy_optimization_loop_test.py
@@ -98,7 +98,7 @@ def test_mock_optimization_loop_no_defaults(
 def test_flaml_optimization_loop(mock_env_no_noise: MockEnv, flaml_opt: MlosCoreOptimizer) -> None:
     """Toy optimization loop with mock environment and FLAML optimizer."""
     (score, tunables) = _optimize(mock_env_no_noise, flaml_opt)
-    assert score == pytest.approx(60.15, 0.01)
+    assert score == pytest.approx(63.77, 0.01)
     assert tunables.get_param_values() == {
         "vmSize": "Standard_B2s",
         "idle": "halt",

diff --git a/mlos_core/setup.py b/mlos_core/setup.py
@@ -68,7 +68,7 @@ def _get_long_desc_from_readme(base_url: str) -> dict:
 
 extra_requires: dict[str, list[str]] = {  # pylint: disable=consider-using-namedtuple-or-dataclass
     "flaml": [
-        "flaml",
+        "flaml>=2.4.0",
         "flaml[blendsearch]",
     ],
     # NOTE: Major refactoring on SMAC and ConfigSpace v1.0 starting from v2.2

diff --git a/setup.cfg b/setup.cfg
@@ -52,6 +52,7 @@ filterwarnings =
     ignore:.*(get_hyperparam|get_dictionary|parents_of|to_vector|(list\(.*values\(\)\))).*:DeprecationWarning:smac:0
     ignore:.*(Please leave at default or explicitly set .size=None).*:DeprecationWarning:smac:0
     ignore:.*(declarative_base.*function is now available as sqlalchemy.orm.declarative_base):DeprecationWarning:optuna:0
+    ignore:.*(IntUniformDistribution.* is deprecated and internally converted to IntDistribution):FutureWarning:optuna:0
     ignore:.*(Trying to register a configuration that was not previously suggested).*:UserWarning:.*llamatune.*:0
     ignore:.*(DISPLAY environment variable is set).*:UserWarning:.*conftest.*:0
     ignore:.*(coroutine 'sleep' was never awaited).*:RuntimeWarning:.*event_loop_context_test.*:0