diff --git a/model_manager/worker/worker.py b/model_manager/worker/worker.py index ed553a4..cb96773 100644 --- a/model_manager/worker/worker.py +++ b/model_manager/worker/worker.py @@ -152,10 +152,8 @@ async def main(): except BaseException as e: # noqa: BLE001 logger.custom_error(f'An unhandled exception occurred: {e}', metadata) finally: - if notification_handler: - notification_handler.shutdown() - if activities: - await activities.shutdown() + notification_handler.shutdown() + await activities.shutdown() # Exit with a non-zero status code to indicate failure to Kubernetes metrics.APP_UP.labels(pod_id=POD_ID).set(0) # Mark app as DOWN sys.exit(1)