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: