[tune] trim kwargs in shim instantiation functions (#12544)

This commit is contained in:
Kaushik B
2020-12-02 12:07:00 -08:00
committed by GitHub
parent da42bf29d0
commit 7422abddb4
4 changed files with 29 additions and 4 deletions
+6 -1
View File
@@ -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__ = [