fix check for env var (#1151)

axolotl-ai-cloud · Jan 19, 2024 · cbecf3e · cbecf3e
1 parent 729740d
commit cbecf3e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/scripts/cloud-entrypoint.sh b/scripts/cloud-entrypoint.sh
@@ -5,17 +5,15 @@ echo "Exporting environment variables..."
 printenv | grep -E '^RUNPOD_|^PATH=|^_=' | sed 's/^\(.*\)=\(.*\)$/export \1="\2"/' >> /etc/rp_environment
 echo 'source /etc/rp_environment' >> ~/.bashrc
 
-if [[ $PUBLIC_KEY ]]
-then
+if [[ $PUBLIC_KEY ]]; then
     # runpod
     mkdir -p ~/.ssh
     chmod 700 ~/.ssh
     echo $PUBLIC_KEY >> ~/.ssh/authorized_keys
     chmod 700 -R ~/.ssh
     # Start the SSH service in the background
     service ssh start
-elif [[ $SSH_KEY ]]
-then
+elif [ -n "$SSH_KEY" ]; then
     # latitude.sh
     mkdir -p ~/.ssh
     chmod 700 ~/.ssh