Skip to content
Snippets Groups Projects

Log a message when everything went well as well

Merged Eva Bardou requested to merge log-yay-message into master
All threads resolved!
3 files
+ 19
13
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -230,12 +230,13 @@ class ElementsWorker(
with contextlib.suppress(Exception):
self.update_activity(element.id, ActivityState.Error)
message = f'Ran on {count} element{"s"[:count>1]}: {count - failed} completed, {failed} failed'
if failed:
logger.error(
f"Ran on {count} elements: {count - failed} completed, {failed} failed"
)
logger.error(message)
if failed >= count: # Everything failed!
sys.exit(1)
else:
logger.info(message)
def process_element(self, element: Element | CachedElement):
"""
@@ -505,9 +506,10 @@ class DatasetWorker(BaseWorker, DatasetMixin, TaskMixin):
if dataset_artifact:
dataset_artifact.unlink(missing_ok=True)
message = f'Ran on {count} dataset{"s"[:count>1]}: {count - failed} completed, {failed} failed'
if failed:
logger.error(
f"Ran on {count} datasets: {count - failed} completed, {failed} failed"
)
logger.error(message)
if failed >= count: # Everything failed!
sys.exit(1)
else:
logger.info(message)
Loading