update jl-wandb to use inputs instead of env

Author: deep-diver

.github/actions/exp/jl-wandb/action.yml

@@ -22,47 +22,54 @@ runs:
   steps:
       - name: prepare script
         env:
-          CUR_BRANCH: $CUR_BRANCH
-          CUR_PR_ID: $CUR_PR_ID
-          GDRIVE_CREDENTIAL: $GDRIVE_CREDENTIAL
-          GH_ACCESS_TOKEN: $GH_ACCESS_TOKEN
-          CLOUD_ID: $JARVISLABS_ID
-          CLOUD_AT: $JARVISLABS_ACCESS_TOKEN
-          WANDB_PROJECT: $WANDB_PROJECT
-          WANDB_API_KEY: $WANDB_API_KEY
+          CUR_BRANCH: ${{ inputs.CUR_BRANCH }}
+          CUR_PR_ID: ${{ inputs.CUR_PR_ID }}
+          GDRIVE_CREDENTIAL: ${{ inputs.GDRIVE_CREDENTIAL }}
+          GH_ACCESS_TOKEN: ${{ inputs.GH_ACCESS_TOKEN }}
+          CLOUD_ID: ${{ inputs.JARVISLABS_ID }}
+          CLOUD_AT: ${{ inputs.JARVISLABS_ACCESS_TOKEN }}
+          WANDB_PROJECT: ${{ inputs.WANDB_PROJECT }}
+          WANDB_API_KEY: ${{ inputs.WANDB_API_KEY }}
         run: |
           envsubst \
             '$CUR_BRANCH, \
              $CUR_PR_ID, \
              $GDRIVE_CREDENTIAL, \
              $GH_ACCESS_TOKEN, \
              $CLOUD_ID, \
-             $CLOUD_AT' \
-             < scripts/js_exp_wandb.sh \
-             > scripts/js_exp_wandb_tmp.sh
+             $CLOUD_AT, \
+             $WANDB_PROJECT, \
+             $WANDB_API_KEY' \
+             < scripts/jl_exp_wandb.sh \
+             > scripts/jl_exp_wandb_tmp.sh
+        shell: bash             
 
       - name: install jarvislabs client
         run: |
           pip install typer
           pip install git+https://github.com/jarvislabsai/jlclient.git
+        shell: bash          
 
       - name: add script to jarvislabs
         id: add_script
         run: |
           python clouds/jarvislabs.py \
             script add \
-              ${{ env.CLOUD_AT }} ${{ env.CLOUD_ID }} scripts/js_exp_wandb_tmp.sh \
+              ${{ inputs.JARVISLABS_ACCESS_TOKEN }} ${{ inputs.JARVISLABS_ID }} scripts/jl_exp_wandb_tmp.sh \
                 > outputs.txt
           echo "::set-output name=script_id::$(cat outputs.txt)"
+        shell: bash          
 
       - name: create vm on jarvislabs
         run: |
           python clouds/jarvislabs.py \
             vm create \
-              ${{ env.CLOUD_AT }} ${{ env.CLOUD_ID }} ${{ steps.add_script.outputs.script_id }}
+              ${{ inputs.JARVISLABS_ACCESS_TOKEN }} ${{ inputs.JARVISLABS_ID }} ${{ steps.add_script.outputs.script_id }}
+        shell: bash              
 
       - name: remove script from jarvislabs
         run: |
           python clouds/jarvislabs.py \
             script remove \
-              ${{ env.CLOUD_AT }} ${{ env.CLOUD_ID }} ${{ steps.add_script.outputs.script_id }}
+              ${{ inputs.JARVISLABS_ACCESS_TOKEN }} ${{ inputs.JARVISLABS_ID }} ${{ steps.add_script.outputs.script_id }}
+        shell: bash