Skip to content

Commit 23ffd98

Browse files
authored
[autorevert] Make parameters configurable via environment variables (#7005)
1 parent e5e0f86 commit 23ffd98

File tree

1 file changed

+2
-8
lines changed
  • aws/lambda/pytorch-auto-revert/pytorch_auto_revert

1 file changed

+2
-8
lines changed

aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -171,16 +171,10 @@ def main(*args, **kwargs) -> None:
171171

172172
if opts.subcommand is None:
173173
autorevert_checker(
174-
[
175-
"Lint",
176-
"trunk",
177-
"pull",
178-
"inductor",
179-
"linux-binary-manywheel",
180-
],
174+
os.environ.get("WORKFLOWS", "Lint,trunk,pull,inductor").split(","),
181175
do_restart=True,
182176
do_revert=True,
183-
hours=2,
177+
hours=int(os.environ.get("HOURS", 48)),
184178
verbose=True,
185179
dry_run=opts.dry_run,
186180
ignore_common_errors=True,

0 commit comments

Comments
 (0)