diff --git a/banzai/main.py b/banzai/main.py index bd1c1fd5..13865113 100755 --- a/banzai/main.py +++ b/banzai/main.py @@ -181,8 +181,9 @@ def start_stacking_scheduler(): schedule_calibration_stacking.s(site=site, runtime_context=vars(runtime_context)), queue=runtime_context.CELERY_TASK_QUEUE_NAME) - app.Beat(schedule='/tmp/celerybeat-schedule', pidfile='/tmp/celerybeat.pid', working_directory='/tmp').run() + beat = celery.bin.beat.beat(app=app) logger.info('Starting celery beat') + beat.run(schedule='/tmp/celerybeat-schedule', pidfile='/tmp/celerybeat.pid', working_directory='/tmp') def run_realtime_pipeline(): extra_console_arguments = [{'args': ['--n-processes'], diff --git a/setup.cfg b/setup.cfg index d21c76a9..8e7c6b29 100755 --- a/setup.cfg +++ b/setup.cfg @@ -61,7 +61,7 @@ install_requires = pytest>=4.0 pyyaml psycopg2-binary - celery[redis]>5,<6 + celery[redis]>=4.3.1,<5 apscheduler python-dateutil ocs_ingester>=3.0.4,<4.0.0