Skip to content

Commit

Permalink
fix: retry_on_transient_error should be int (#113)
Browse files Browse the repository at this point in the history
Co-authored-by: Han Wang <wang_han@iapcm.ac.cn>
  • Loading branch information
wanghan-iapcm and Han Wang committed Jan 7, 2023
1 parent c0e6b4c commit 1261939
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions dpgen2/utils/step_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,13 +41,13 @@ def variant_executor():
def template_conf_args():
doc_image = 'The image to run the step.'
doc_timeout = 'The time limit of the OP. Unit is second.'
doc_retry_on_transient_error = 'Retry the step if a TransientError is raised.'
doc_retry_on_transient_error = 'The number of retry times if a TransientError is raised.'
doc_timeout_as_transient_error = 'Treat the timeout as TransientError.'
doc_envs = 'The environmental variables.'
return [
Argument("image", str, optional=True, default=default_image, doc=doc_image),
Argument("timeout", int, optional=True, default=None, doc=doc_timeout),
Argument("retry_on_transient_error", bool, optional=True, default=None, doc=doc_retry_on_transient_error),
Argument("retry_on_transient_error", int, optional=True, default=None, doc=doc_retry_on_transient_error),
Argument("timeout_as_transient_error", bool, optional=True, default=False, doc=doc_timeout_as_transient_error),
Argument("envs", dict, optional=True, default=None, doc=doc_envs),
]
Expand Down

0 comments on commit 1261939

Please sign in to comment.