Merge branch 'tasks-dec' of https://github.com/GoogleCloudPlatform/gc…

…e-rescue into tasks-dec
GoogleCloudPlatform · Feb 28, 2024 · c3adce6 · c3adce6
2 parents 9bb3889 + 13a0c1c
commit c3adce6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/gce_rescue/utils.py b/gce_rescue/utils.py
@@ -151,9 +151,8 @@ def inner(*args, **kwargs):
     _logger.info('task %s '
                  'started at: %s',
                  task_func.__name__, datetime.now().strftime('%H:%M:%S'))
-    task_index = kwargs.pop('task_index') if kwargs.get('task_index') else None
-    total_tasks = kwargs.pop('total_tasks') if kwargs.get('total_tasks') \
-      else None
+    task_index = kwargs.pop('task_index', None)
+    total_tasks = kwargs.pop('total_tasks', None)
     try:
       res = task_func(*args, **kwargs)
     except HttpError as e: