Fixes to calling the stacking task converting the context.

LCOGT · Mar 22, 2024 · b00844d · b00844d
1 parent c5f6d84
commit b00844d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/banzai_floyds/main.py b/banzai_floyds/main.py
@@ -10,6 +10,7 @@
 from banzai.utils.date_utils import TIMESTAMP_FORMAT
 import datetime
 from astropy.time import Time
+from banzai.context import Context
 
 
 logger = logging.getLogger('banzai')
@@ -60,8 +61,13 @@ def populate_photometric_standards():
 
 @app.task(name='celery.stack_flats', reject_on_worker_lost=True, max_retries=5)
 def stack_flats_task(min_date, max_date, instrument_id, runtime_context):
-    instrument = banzai.dbs.get_instrument_by_id(instrument_id, db_address=runtime_context.db_address)
-    calibrations.make_master_calibrations(instrument, 'LAMPFLAT', min_date, max_date, runtime_context)
+    try:
+        runtime_context = Context(runtime_context)
+        instrument = banzai.dbs.get_instrument_by_id(instrument_id, db_address=runtime_context.db_address)
+        calibrations.make_master_calibrations(instrument, 'LAMPFLAT', min_date, max_date, runtime_context)
+    except Exception:
+        logger.error("Exception processing frame: {error}".format(error=logs.format_exception()),
+                     extra_tags={'instrument_id': instrument_id, 'min_date': min_date, 'max_date': max_date})
 
 
 def banzai_floyds_stack_flats():

diff --git a/helm-chart/banzai-floyds/templates/workers.yaml b/helm-chart/banzai-floyds/templates/workers.yaml
@@ -29,7 +29,7 @@ spec:
           command:
             - "celery"
             - "-A"
-            - "banzai"
+            - "banzai_floyds.main:app"
             - "worker"
             - "--concurrency"
             - "1"