From a0767b7fdddf078a0fb9041bf437022ae70a4c1f Mon Sep 17 00:00:00 2001 From: Max Halford Date: Mon, 22 Jul 2024 19:46:56 +0200 Subject: [PATCH] ruff --- lea/clients/bigquery.py | 2 +- lea/runner.py | 7 +++---- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/lea/clients/bigquery.py b/lea/clients/bigquery.py index 3daa608..daf1df2 100644 --- a/lea/clients/bigquery.py +++ b/lea/clients/bigquery.py @@ -40,7 +40,7 @@ def client(self): return bigquery.Client( # TODO: enable this so the project ID from the credentials is used, which makes sense # when the client is located in the project doing the compute (with slots allocated) - #project=self.credentials.project_id, + # project=self.credentials.project_id, project=self.project_id, credentials=self.credentials, location=self.location, diff --git a/lea/runner.py b/lea/runner.py index ef35a4b..afea03c 100644 --- a/lea/runner.py +++ b/lea/runner.py @@ -299,10 +299,9 @@ def get_status(view_key): not_done = [view_key for view_key in execution_order if view_key not in cache] statuses = {view_key: get_status(view_key) for view_key in not_done} - not_done = ( - [view_key for view_key in not_done if statuses[view_key] != RUNNING] + - [view_key for view_key in not_done if statuses[view_key] == RUNNING] - ) + not_done = [view_key for view_key in not_done if statuses[view_key] != RUNNING] + [ + view_key for view_key in not_done if statuses[view_key] == RUNNING + ] for i, view_key in list(enumerate(not_done, start=1))[-show:]: status = statuses[view_key] duration = (