fix: syne-tune random search import

whittle-org · Feb 13, 2025 · 5f8a0a2 · 5f8a0a2
1 parent 0aab4d9
commit 5f8a0a2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/whittle/search/stratified_search.py b/whittle/search/stratified_search.py
@@ -3,7 +3,9 @@
 from typing import Any
 
 from syne_tune.optimizer.schedulers import FIFOScheduler
-from syne_tune.optimizer.schedulers.searchers.random_grid_searcher import RandomSearcher
+from syne_tune.optimizer.schedulers.searchers import (
+    LegacyRandomSearcher as RandomSearcher,
+)
 
 from whittle.sampling.param_bins import ParamBins