Skip to content

Commit 88b60bb

Browse files
alan-z-yulint-actionldoshi
authored
Fixing type issue (#225)
* Fixing type issue * Fixing type issue in experiment_runner * Fix code style issues with Black --------- Co-authored-by: Lint Action <lint-action@samuelmeuli.com> Co-authored-by: ldoshi <lyric@mit.edu>
1 parent 1c9cb7c commit 88b60bb

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tools/experiment_runner.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def _executor(
7373
experiment_name_prefix: str,
7474
args: list[Any],
7575
sweep_keys: list[str],
76-
iteration_values: list[Any],
76+
iteration_values: tuple[Any],
7777
) -> list[str]:
7878

7979
iteration_args = copy.deepcopy(args)
@@ -134,7 +134,12 @@ def run_experiments(
134134

135135
if sweep_keys:
136136
executor_fn = functools.partial(
137-
_executor, execute_fn, experiment_name_prefix, args, sweep_keys
137+
138+
139+
140+
141+
142+
, execute_fn, experiment_name_prefix, args, sweep_keys
138143
)
139144
with Pool(num_processes) as pool:
140145
# Iterate sweep combinations.

0 commit comments

Comments
 (0)