Skip to content

Commit

Permalink
Lint fixes for #421
Browse files Browse the repository at this point in the history
Similar to 8cb41b8. A couple new fixes
now that more code has been landed before the CI check went up.
Separating this out from #421 to try and keep the commit history
somewhat clean.
  • Loading branch information
boomanaiden154 committed Jan 29, 2025
1 parent 59a5478 commit 1851281
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 9 deletions.
14 changes: 6 additions & 8 deletions compiler_opt/es/blackbox_evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,14 +124,12 @@ def __init__(self, train_corpus: corpus.Corpus,
def get_results(
self, pool: FixedWorkerPool, perturbations: List[policy_saver.Policy]
) -> List[concurrent.futures.Future]:
job_args = []
for perturbation in perturbations:
job_args.append({
'modules': self._train_corpus.module_specs,
'function_index_path': self._function_index_path,
'bb_trace_path': self._bb_trace_path,
'tflite_policy': perturbation
})
job_args = [{
'modules': self._train_corpus.module_specs,
'function_index_path': self._function_index_path,
'bb_trace_path': self._bb_trace_path,
'tflite_policy': perturbation
} for perturbation in perturbations]

_, futures = buffered_scheduler.schedule_on_worker_pool(
action=lambda w, args: w.compile_corpus_and_evaluate(**args),
Expand Down
2 changes: 1 addition & 1 deletion compiler_opt/es/blackbox_test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ def compile(self, policy: policy_saver.Policy,

class ESTraceWorker(worker.Worker):
"""Temporary placeholder worker.
This is a test worker for TraceBlackboxEvaluator that expects a slightly
different interface than other workers.
"""
Expand Down

0 comments on commit 1851281

Please sign in to comment.