From 362ea0dc38461e278d65e7d4f49bd5329e0385a4 Mon Sep 17 00:00:00 2001
From: Edward Foyle <foyleef@amazon.com>
Date: Wed, 28 Jul 2021 19:30:12 -0700
Subject: [PATCH] fix: correct behavior of --yes on push when missing env var

---
 .../awscloudformation/utils/environmentVariablesHelper.ts       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/amplify-category-function/src/provider-utils/awscloudformation/utils/environmentVariablesHelper.ts b/packages/amplify-category-function/src/provider-utils/awscloudformation/utils/environmentVariablesHelper.ts
index 6bb5887503f..a9d5ffcf049 100644
--- a/packages/amplify-category-function/src/provider-utils/awscloudformation/utils/environmentVariablesHelper.ts
+++ b/packages/amplify-category-function/src/provider-utils/awscloudformation/utils/environmentVariablesHelper.ts
@@ -159,7 +159,7 @@ export const askEnvironmentVariableCarryOut = async (
 };
 
 export const ensureEnvironmentVariableValues = async (context: $TSContext) => {
-  const yesFlagSet = context?.exeInfo?.inputParams?.yes;
+  const yesFlagSet = context?.exeInfo?.inputParams?.yes || context?.input?.options?.yes;
   const currentEnvName = stateManager.getLocalEnvInfo()?.envName;
   const teamProviderInfo = stateManager.getTeamProviderInfo(undefined, {
     throwIfNotExist: false,