Skip to content

Commit 44f0633

Browse files
Kiuk Chungfacebook-github-bot
authored andcommitted
(torchx/cli) allow the default scheduler to be specified via .torchxconfig (#506)
Summary: Pull Request resolved: #506 Lets users specify their choice of default scheduler via `.torchxconfig` (similar to how they can specify the default component in `.torchxconfig`) Reviewed By: d4l3k, dragonxlwang Differential Revision: D36849694 fbshipit-source-id: 9d32d928629a2e7f9b781ef76eba050fa6b43060
1 parent 8672dca commit 44f0633

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchx/cli/cmd_run.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,9 @@ def add_arguments(self, subparser: argparse.ArgumentParser) -> None:
122122
"-s",
123123
"--scheduler",
124124
type=str,
125-
help=f"Name of the scheduler to use. One of: [{','.join(scheduler_names)}]",
126125
default=get_default_scheduler_name(),
126+
action=torchxconfig_run,
127+
help=f"Name of the scheduler to use. One of: [{','.join(scheduler_names)}]",
127128
)
128129
subparser.add_argument(
129130
"-cfg",

0 commit comments

Comments
 (0)