From 42eb69799b585ffa27b6e51aba6353b1c6447a45 Mon Sep 17 00:00:00 2001
From: Forrest Collman <forrest.collman@gmail.com>
Date: Thu, 22 Jan 2026 13:23:23 -0800
Subject: [PATCH] adding retries to delete task to handle info service being
 down

---
 .../workflows/periodic_database_removal.py                  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/materializationengine/workflows/periodic_database_removal.py b/materializationengine/workflows/periodic_database_removal.py
index 8e5af9f4..ccaf7f82 100644
--- a/materializationengine/workflows/periodic_database_removal.py
+++ b/materializationengine/workflows/periodic_database_removal.py
@@ -47,7 +47,11 @@ def get_valid_versions(session):
     return [str(version) for version in valid_versions]
 
 
-@celery.task(name="workflow:remove_expired_databases")
+@celery.task(name="workflow:remove_expired_databases",
+             bind=True,
+             acks_late=True,
+             autoretry_for=(Exception,),
+             max_retries=3)
 def remove_expired_databases(delete_threshold: int = 5, datastack: str = None) -> str:
     """Remove expired databases and clean up their metadata.