mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2025-12-30 21:29:44 -06:00
e4b1d36de6
* fix: child index on retry not being reset correctly * chore: ver * chore: changelog * feat: test
44 lines
948 B
Python
44 lines
948 B
Python
import argparse
|
|
from typing import cast
|
|
|
|
from hatchet_sdk import Hatchet
|
|
from tests.child_spawn_cache_on_retry.worker import (
|
|
spawn_cache_on_retry_child,
|
|
spawn_cache_on_retry_parent,
|
|
)
|
|
from tests.correct_failure_on_timeout_with_multi_concurrency.workflow import (
|
|
multiple_concurrent_cancellations_test_workflow,
|
|
)
|
|
|
|
hatchet = Hatchet(debug=True)
|
|
|
|
|
|
def main(slots: int) -> None:
|
|
worker = hatchet.worker(
|
|
"e2e-test-worker-2",
|
|
slots=slots,
|
|
workflows=[
|
|
multiple_concurrent_cancellations_test_workflow,
|
|
spawn_cache_on_retry_parent,
|
|
spawn_cache_on_retry_child,
|
|
],
|
|
)
|
|
|
|
worker.start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--slots",
|
|
type=int,
|
|
default=100,
|
|
required=False,
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
slots = cast(int | None, args.slots) or 100
|
|
|
|
main(slots)
|