stream3r/worker/__init__.py

"""Worker utilities for running STream3R jobs via RQ."""

from rq.queue import Queue

from .config import WorkerSettings

_settings = WorkerSettings.from_env()
if _settings.default_job_timeout and _settings.default_job_timeout > 0:
    Queue.DEFAULT_TIMEOUT = _settings.default_job_timeout

from .tasks import keyframe_selection_job, model_build_job, pose_pointmap_job  # noqa: E402

__all__ = [
    "pose_pointmap_job",
    "model_build_job",
    "keyframe_selection_job",
]