Skip to content

Commit 9fe7afa

Browse files
committed
Update get_registry script
1 parent 31108ed commit 9fe7afa

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

crates/fluxqueue-worker/scripts/get_registry.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,14 @@ def get_registry( # noqa: C901
1919
registry = {"tasks": {}, "contexts": {}}
2020
for _name, obj in inspect.getmembers(module):
2121
if inspect.isfunction(obj):
22+
is_fluxqueue_task = getattr(obj, "fluxqueue", False)
23+
24+
if not is_fluxqueue_task:
25+
continue
26+
2227
task_name = getattr(obj, "task_name", None)
2328
task_queue = getattr(obj, "queue", None)
29+
2430
if not task_queue or task_queue != queue:
2531
continue
2632

0 commit comments

Comments
 (0)