@@ -255,6 +255,7 @@ def taskworker_scheduler(redis_cluster: str, **options: Any) -> None:
255
255
"""
256
256
from django .conf import settings
257
257
258
+ from sentry import options as featureflags
258
259
from sentry .taskworker .registry import taskregistry
259
260
from sentry .taskworker .scheduler .runner import RunStorage , ScheduleRunner
260
261
from sentry .utils .redis import redis_clusters
@@ -266,7 +267,7 @@ def taskworker_scheduler(redis_cluster: str, **options: Any) -> None:
266
267
267
268
with managed_bgtasks (role = "taskworker-scheduler" ):
268
269
runner = ScheduleRunner (taskregistry , run_storage )
269
- enabled_schedules = set (options .get ("taskworker.scheduler.rollout" , []))
270
+ enabled_schedules = set (featureflags .get ("taskworker.scheduler.rollout" , []))
270
271
for key , schedule_data in settings .TASKWORKER_SCHEDULES .items ():
271
272
if key in enabled_schedules :
272
273
runner .add (key , schedule_data )
@@ -479,6 +480,8 @@ def cron(**options: Any) -> None:
479
480
"Run periodic task dispatcher."
480
481
from django .conf import settings
481
482
483
+ from sentry import options as featureflags
484
+
482
485
if settings .CELERY_ALWAYS_EAGER :
483
486
raise click .ClickException (
484
487
"Disable CELERY_ALWAYS_EAGER in your settings file to spawn workers."
@@ -488,7 +491,7 @@ def cron(**options: Any) -> None:
488
491
489
492
old_schedule = app .conf .CELERYBEAT_SCHEDULE
490
493
new_schedule = {}
491
- task_schedules = set (options .get ("taskworker.scheduler.rollout" , []))
494
+ task_schedules = set (featureflags .get ("taskworker.scheduler.rollout" , []))
492
495
for key , schedule_data in old_schedule .items ():
493
496
if key not in task_schedules :
494
497
new_schedule [key ] = schedule_data
0 commit comments