Skip to content
This repository was archived by the owner on Feb 20, 2025. It is now read-only.

Commit fdd7e95

Browse files
authored
fix: make all examples runnable (hatchet-dev#302)
1 parent e3a4387 commit fdd7e95

File tree

7 files changed

+58
-24
lines changed

7 files changed

+58
-24
lines changed

examples/cancellation/worker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,13 @@ async def step1(self, context: Context) -> None:
2323
print("Cancelled")
2424

2525

26-
workflow = CancelWorkflow()
27-
worker = hatchet.worker("cancellation-worker", max_runs=4)
28-
worker.register_workflow(workflow)
26+
def main() -> None:
27+
workflow = CancelWorkflow()
28+
worker = hatchet.worker("cancellation-worker", max_runs=4)
29+
worker.register_workflow(workflow)
2930

30-
worker.start()
31+
worker.start()
32+
33+
34+
if __name__ == "__main__":
35+
main()

examples/concurrency_limit_rr/worker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,13 @@ def step1(self, context: Context) -> None:
3333
pass
3434

3535

36-
workflow = ConcurrencyDemoWorkflowRR()
37-
worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10)
38-
worker.register_workflow(workflow)
36+
def main() -> None:
37+
workflow = ConcurrencyDemoWorkflowRR()
38+
worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10)
39+
worker.register_workflow(workflow)
3940

40-
worker.start()
41+
worker.start()
42+
43+
44+
if __name__ == "__main__":
45+
main()

examples/delayed/worker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,13 @@ def step1(self, context: Context) -> None:
3232
print(f"message \t {context.workflow_input()['message']}")
3333

3434

35-
worker = hatchet.worker("delayed-worker", max_runs=4)
36-
worker.register_workflow(PrintSchedule())
37-
worker.register_workflow(PrintPrinter())
35+
def main() -> None:
36+
worker = hatchet.worker("delayed-worker", max_runs=4)
37+
worker.register_workflow(PrintSchedule())
38+
worker.register_workflow(PrintPrinter())
3839

39-
worker.start()
40+
worker.start()
41+
42+
43+
if __name__ == "__main__":
44+
main()

examples/manual_trigger/worker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,13 @@ def step2(self, context: Context) -> dict[str, str]:
4747
return {"step2": "data2"}
4848

4949

50-
workflow = ManualTriggerWorkflow()
51-
worker = hatchet.worker("manual-worker", max_runs=4)
52-
worker.register_workflow(workflow)
50+
def main() -> None:
51+
workflow = ManualTriggerWorkflow()
52+
worker = hatchet.worker("manual-worker", max_runs=4)
53+
worker.register_workflow(workflow)
5354

54-
worker.start()
55+
worker.start()
56+
57+
58+
if __name__ == "__main__":
59+
main()

examples/overrides/worker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,13 @@ def step4(self, context: Context) -> dict[str, str]:
6969
}
7070

7171

72-
workflow = OverridesWorkflow()
73-
worker = hatchet.worker("overrides-worker")
74-
worker.register_workflow(workflow)
72+
def main() -> None:
73+
workflow = OverridesWorkflow()
74+
worker = hatchet.worker("overrides-worker")
75+
worker.register_workflow(workflow)
7576

76-
worker.start()
77+
worker.start()
78+
79+
80+
if __name__ == "__main__":
81+
main()

examples/rate_limit/worker.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,3 +23,7 @@ def main() -> None:
2323
worker.register_workflow(RateLimitWorkflow())
2424

2525
worker.start()
26+
27+
28+
if __name__ == "__main__":
29+
main()

examples/sticky_workers/worker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,12 @@ def child(self, context: Context) -> dict[str, str | None]:
3636
return {"worker": context.worker.id()}
3737

3838

39-
worker = hatchet.worker("sticky-worker", max_runs=10)
40-
worker.register_workflow(StickyWorkflow())
41-
worker.register_workflow(StickyChildWorkflow())
42-
worker.start()
39+
def main() -> None:
40+
worker = hatchet.worker("sticky-worker", max_runs=10)
41+
worker.register_workflow(StickyWorkflow())
42+
worker.register_workflow(StickyChildWorkflow())
43+
worker.start()
44+
45+
46+
if __name__ == "__main__":
47+
main()

0 commit comments

Comments
 (0)