set envirornment variable as PRODUCTION if the stage type is DEFAULT

pinterest · Dec 13, 2023 · f3b697c · f3b697c
1 parent dc7519c
commit f3b697c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deploy-agent/deployd/common/config.py b/deploy-agent/deployd/common/config.py
@@ -105,7 +105,7 @@ def update_variables(self, deploy_status):
         if deploy_status.report.stageName:
             self._environ['STAGE_NAME'] = deploy_status.report.stageName
         if deploy_status.report.stageType:
-            self._environ['COMPUTE_ENV_TYPE'] = deploy_status.report.stageType
+            self._environ['COMPUTE_ENV_TYPE'] = "PRODUCTION" if deploy_status.report.stageType == "DEFAULT" else deploy_status.report.stageType
         if deploy_status.first_deploy:
             self._environ['FIRST_DEPLOY'] = str(deploy_status.first_deploy)
         if deploy_status.is_docker: