gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1241: Simplify worker shutdown logic by removing conditional checks.

Browse Source
This commit is contained in:
Bruno Domingues
2025-10-30 15:14:43 -03:00
parent f5c1ac3615
commit bb25a19d8c
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
model_manager/worker/worker.py
View File

@@ -152,10 +152,8 @@ async def main():
except BaseException as e: # noqa: BLE001 except BaseException as e: # noqa: BLE001
logger.custom_error(f'An unhandled exception occurred: {e}', metadata) logger.custom_error(f'An unhandled exception occurred: {e}', metadata)
finally: finally:
if notification_handler: notification_handler.shutdown()
notification_handler.shutdown() await activities.shutdown()
if activities:
await activities.shutdown()
# Exit with a non-zero status code to indicate failure to Kubernetes # 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 metrics.APP_UP.labels(pod_id=POD_ID).set(0) # Mark app as DOWN
sys.exit(1) sys.exit(1)