hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Workflow] Fix flaky example(#26960)

Browse source 
Signed-off-by: Siyuan Zhuang <suquark@gmail.com>
This commit is contained in:
Siyuan (Ryans) Zhuang 2022-07-25 12:00:45 -07:00 • committed by GitHub
parent e8222ff600
commit 0e1b77d52a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/ray/workflow/examples/comparisons/argo/retry_workflow.py
View file

@ -38,6 +38,8 @@ def custom_retry_strategy(func: Any, num_retries: int, delay_s: int) -> str:
if __name__ == "__main__":
# Default retry strategy.
print(workflow.run(flaky_step.options(max_retries=10).bind()))
print(
workflow.run(flaky_step.options(max_retries=10, retry_exceptions=True).bind())
)
# Custom strategy.
print(workflow.run(custom_retry_strategy.bind(flaky_step, 10, 1)))