diff --git a/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py b/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py index 5356731230..31e881a0ce 100755 --- a/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py +++ b/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py @@ -60,7 +60,17 @@ def __init__(self): self.secret_store_name = os.environ.get("SECRET_STORE_NAME", "") self.workflows = os.environ.get( "WORKFLOWS", - ",".join(["Lint", "trunk", "pull", "inductor", "linux-aarch64", "slow"]), + ",".join( + [ + "Lint", + "trunk", + "pull", + "inductor", + "linux-aarch64", + "slow", + "vllm-test", + ] + ), ).split(",") def to_autorevert_v2_params(