From 6a4ebc0edb888e97607bd6da6194120516e57b65 Mon Sep 17 00:00:00 2001
From: "pixeebot[bot]" <104101892+pixeebot[bot]@users.noreply.github.com>
Date: Tue, 20 Feb 2024 15:55:24 -0500
Subject: [PATCH] Use Generator Expressions Instead of List Comprehensions
(#38)
Co-authored-by: pixeebot[bot] <104101892+pixeebot[bot]@users.noreply.github.com>
---
sigoptlite/best_assignments.py | 2 +-
sigoptlite/broker.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/sigoptlite/best_assignments.py b/sigoptlite/best_assignments.py
index 0f285f2..a25c6b5 100644
--- 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
index c1f8345..650e2b1 100644
--- 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,