mirror of
https://github.com/wassname/ray.git
synced 2026-07-28 11:25:04 +08:00
[tune] trim kwargs in shim instantiation functions (#12544)
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
from ray.utils import get_function_args
|
||||
from ray.tune.suggest.search import SearchAlgorithm
|
||||
from ray.tune.suggest.basic_variant import BasicVariantGenerator
|
||||
from ray.tune.suggest.suggestion import Searcher, ConcurrencyLimiter
|
||||
@@ -93,7 +94,11 @@ def create_searcher(
|
||||
f"Got: {search_alg}")
|
||||
|
||||
SearcherClass = SEARCH_ALG_IMPORT[search_alg]()
|
||||
return SearcherClass(**kwargs)
|
||||
|
||||
search_alg_args = get_function_args(SearcherClass)
|
||||
trimmed_kwargs = {k: v for k, v in kwargs.items() if k in search_alg_args}
|
||||
|
||||
return SearcherClass(**trimmed_kwargs)
|
||||
|
||||
|
||||
__all__ = [
|
||||
|
||||
Reference in New Issue
Block a user