Skip to content

Commit f122450

Browse files
authored
Merge pull request #106 from misrasaurabh1/codeflash/optimize-run_search-m8ev3hua
2 parents b53882b + da674b9 commit f122450

1 file changed

Lines changed: 4 additions & 8 deletions

File tree

src/hyperactive/run_search.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,12 @@ def proxy(args):
2424

2525

2626
def _get_distribution(distribution):
27-
if hasattr(distribution, "__call__"):
27+
if callable(distribution):
2828
return (distribution, _process_), {}
2929

3030
elif isinstance(distribution, dict):
31-
dist_key = list(distribution.keys())[0]
32-
dist_paras = list(distribution.values())[0]
31+
dist_key = next(iter(distribution))
32+
dist_paras = distribution[dist_key]
3333

3434
return dist_dict[dist_key], dist_paras
3535

@@ -38,11 +38,7 @@ def _get_distribution(distribution):
3838

3939

4040
def run_search(opt_pros, distribution, n_processes):
41-
opts = list(opt_pros.values())
42-
43-
processes = list(opt_pros.keys())
44-
optimizers = list(opt_pros.values())
45-
process_infos = list(zip(processes, optimizers))
41+
process_infos = list(opt_pros.items())
4642

4743
if n_processes == "auto":
4844
n_processes = len(process_infos)

0 commit comments

Comments
 (0)