in ssiog/training.py [0:0]
def main():
logger.info("testing....logger")
try:
success = True
training()
except Exception as e:
success = False
logger.error(f"Workload failed with error: {e}")
traceback.print_exc()
sys.exit(1)
finally:
# Make sure the flush the metrics in the buffer.
close_metrics_logger()
td.destroy_process_group()
if success:
logger.info("Workload completed successfully.")
sys.exit(0)