From cf3efa62e82870a7ec942ca63d9dc16a5743fd43 Mon Sep 17 00:00:00 2001
From: Rajesh Rajendran <rjshrjndrn@gmail.com>
Date: Fri, 14 Jan 2022 06:47:12 +0530
Subject: [PATCH] chore(install): check for -pr in releases

Signed-off-by: Rajesh Rajendran <rjshrjndrn@gmail.com>
---
 scripts/helm/kube-install.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/helm/kube-install.sh b/scripts/helm/kube-install.sh
index 0a42416bf..056a45de4 100755
--- a/scripts/helm/kube-install.sh
+++ b/scripts/helm/kube-install.sh
@@ -147,7 +147,7 @@ function enterprise(){
     enterprise=1
     sed -i "s#enterprise_edition_license.*#enterprise_edition_license: \"${1}\"#g" vars.yaml
     # Updating image version to be ee
-    sed -i "s/\(image_tag.*[0-9]\)\"$/\1-ee\"/" vars.yaml
+    sed 's/\(image_tag.*[0-9]\)\(-pr\)\?"$/\1\2-ee"/' vars.yaml
     echo "Importing enterprise code..."
     cp -rf ../../ee/scripts/* ../
 }