Use Generator Expressions Instead of List Comprehensions (#38)

Co-authored-by: pixeebot[bot] <104101892+pixeebot[bot]@users.noreply.github.com>
sigopt · Feb 20, 2024 · 6a4ebc0 · 6a4ebc0
1 parent 011790e
commit 6a4ebc0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sigoptlite/best_assignments.py b/sigoptlite/best_assignments.py
@@ -12,7 +12,7 @@ def __init__(self, experiment):
     self.experiment = experiment
     self.full_task_cost = None
     if self.experiment.is_multitask:
-      self.full_task_cost = max([task.cost for task in self.experiment.tasks])
+      self.full_task_cost = max(task.cost for task in self.experiment.tasks)
 
   def observation_is_valid_and_full_cost(self, observation):
     if observation.failed:

diff --git a/sigoptlite/broker.py b/sigoptlite/broker.py
@@ -45,7 +45,7 @@ def experiment_progress_dict(self):
     observation_count = len(self.observations)
     observation_budget_consumed = observation_count
     if self.experiment.is_multitask:
-      observation_budget_consumed = sum([o.task.cost for o in self.observations])
+      observation_budget_consumed = sum(o.task.cost for o in self.observations)
 
     return dict(
       observation_budget_consumed=observation_budget_consumed,