dvc_task_celery_app_quickstart_with_temporary_worker.py

1from dvc_task.app import CeleryApp
2from dvc_task.worker import TemporaryWorker
3
4# Initialize the dvc-task Celery application
5app = CeleryApp("dvc-task-example", broker="redis://localhost:6379/0")
6
7@app.task
8def add(x, y):
9    return x + y
10
11if __name__ == "__main__":
12    # Start a temporary worker and execute a task
13    with TemporaryWorker(app):
14        result = add.delay(4, 4)
15        print(f"Task result: {result.get(timeout=10)}")