Fix merge

Author: aybchan

.github/actions/gke-xpk/action.yml

@@ -207,11 +207,6 @@ runs:
       # set container commands in-line
       CMD=$(echo ${PRELUDE} ${CMD} ${POSTLUDE} | sed 's/\n/\ /g')
       echo "CMD=${CMD}" >> ${GITHUB_ENV}
-      PRELUDE=$(echo ${PRELUDE} | sed 's/\n/\ /g')
-      POSTLUDE=$(echo ${POSTLUDE} | sed 's/\n/\ /g')
-      CMD=$(echo ${CMD} | sed 's/\n/\ /g')
-
-      echo "CMD=${PRELUDE} ${CMD} ${POSTLUDE}" >> ${GITHUB_ENV}
 
   - name: Create workload on cluster with XPK
     shell: bash -x -e -u {0}
@@ -234,7 +229,9 @@ runs:
       )
 
       version_geq() {
-        if [[ $(echo -e "$1\n$2" | sort -V | head -n 1) != "$1" ]]; then
+        if [[ $1 == $2 ]]; then
+          return 0
+        elif [[ $(echo -e "$1\n$2" | sort -V | head -n 1) != "$1" ]]; then
           return 0
         fi
         return 1
@@ -246,6 +243,13 @@ runs:
               --env="JAX_COORDINATOR_PORT=3389"
               --env="JAX_COORDINATOR_ADDRESS=\$(JOBSET_NAME)-\$(REPLICATED_JOB_NAME)-0-0.\$(JOBSET_NAME):3389"
           )
+
+        envs_flat=$(echo "${{ inputs.ENVS }}" | tr '\n' ' ')
+        IFS=';' read -ra env_vars <<< "${envs_flat}"
+        for env in "${env_vars[@]}"; do
+            env=$(echo "${env}" | xargs)
+            [[ -n "${env}" ]] && args+=(--env="${env}")
+        done
       fi
 
       if version_geq "${{ inputs.XPK_VERSION }}" "v1.0.0"; then