diff --git a/examples/cancellation/worker.py b/examples/cancellation/worker.py index 7a0e8b35..17716602 100644 --- a/examples/cancellation/worker.py +++ b/examples/cancellation/worker.py @@ -23,8 +23,13 @@ async def step1(self, context: Context) -> None: print("Cancelled") -workflow = CancelWorkflow() -worker = hatchet.worker("cancellation-worker", max_runs=4) -worker.register_workflow(workflow) +def main() -> None: + workflow = CancelWorkflow() + worker = hatchet.worker("cancellation-worker", max_runs=4) + worker.register_workflow(workflow) -worker.start() + worker.start() + + +if __name__ == "__main__": + main() diff --git a/examples/concurrency_limit_rr/worker.py b/examples/concurrency_limit_rr/worker.py index c1a9d273..58722682 100644 --- a/examples/concurrency_limit_rr/worker.py +++ b/examples/concurrency_limit_rr/worker.py @@ -33,8 +33,13 @@ def step1(self, context: Context) -> None: pass -workflow = ConcurrencyDemoWorkflowRR() -worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10) -worker.register_workflow(workflow) +def main() -> None: + workflow = ConcurrencyDemoWorkflowRR() + worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10) + worker.register_workflow(workflow) -worker.start() + worker.start() + + +if __name__ == "__main__": + main() diff --git a/examples/delayed/worker.py b/examples/delayed/worker.py index aa47f5c9..cea0b838 100644 --- a/examples/delayed/worker.py +++ b/examples/delayed/worker.py @@ -32,8 +32,13 @@ def step1(self, context: Context) -> None: print(f"message \t {context.workflow_input()['message']}") -worker = hatchet.worker("delayed-worker", max_runs=4) -worker.register_workflow(PrintSchedule()) -worker.register_workflow(PrintPrinter()) +def main() -> None: + worker = hatchet.worker("delayed-worker", max_runs=4) + worker.register_workflow(PrintSchedule()) + worker.register_workflow(PrintPrinter()) -worker.start() + worker.start() + + +if __name__ == "__main__": + main() diff --git a/examples/manual_trigger/worker.py b/examples/manual_trigger/worker.py index 75e56233..9cf1eef5 100644 --- a/examples/manual_trigger/worker.py +++ b/examples/manual_trigger/worker.py @@ -47,8 +47,13 @@ def step2(self, context: Context) -> dict[str, str]: return {"step2": "data2"} -workflow = ManualTriggerWorkflow() -worker = hatchet.worker("manual-worker", max_runs=4) -worker.register_workflow(workflow) +def main() -> None: + workflow = ManualTriggerWorkflow() + worker = hatchet.worker("manual-worker", max_runs=4) + worker.register_workflow(workflow) -worker.start() + worker.start() + + +if __name__ == "__main__": + main() diff --git a/examples/overrides/worker.py b/examples/overrides/worker.py index 9302d46a..1af30f1e 100644 --- a/examples/overrides/worker.py +++ b/examples/overrides/worker.py @@ -69,8 +69,13 @@ def step4(self, context: Context) -> dict[str, str]: } -workflow = OverridesWorkflow() -worker = hatchet.worker("overrides-worker") -worker.register_workflow(workflow) +def main() -> None: + workflow = OverridesWorkflow() + worker = hatchet.worker("overrides-worker") + worker.register_workflow(workflow) -worker.start() + worker.start() + + +if __name__ == "__main__": + main() diff --git a/examples/rate_limit/worker.py b/examples/rate_limit/worker.py index fb3cd3ac..97c19321 100644 --- a/examples/rate_limit/worker.py +++ b/examples/rate_limit/worker.py @@ -23,3 +23,7 @@ def main() -> None: worker.register_workflow(RateLimitWorkflow()) worker.start() + + +if __name__ == "__main__": + main() diff --git a/examples/sticky_workers/worker.py b/examples/sticky_workers/worker.py index abedb820..266a5706 100644 --- a/examples/sticky_workers/worker.py +++ b/examples/sticky_workers/worker.py @@ -36,7 +36,12 @@ def child(self, context: Context) -> dict[str, str | None]: return {"worker": context.worker.id()} -worker = hatchet.worker("sticky-worker", max_runs=10) -worker.register_workflow(StickyWorkflow()) -worker.register_workflow(StickyChildWorkflow()) -worker.start() +def main() -> None: + worker = hatchet.worker("sticky-worker", max_runs=10) + worker.register_workflow(StickyWorkflow()) + worker.register_workflow(StickyChildWorkflow()) + worker.start() + + +if __name__ == "__main__": + main()