minor modifications to gcp_auth

ivy-llc · Oct 6, 2023 · b08ed07 · b08ed07
1 parent 8109adc
commit b08ed07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/.github/auth/vm_auth.py b/.github/auth/vm_auth.py
@@ -14,8 +14,6 @@ def authenticate_vm(path):
 
 def _start_ssh_session(response, creds, username, passphrase):
     external_ip = response["networkInterfaces"][0]["accessConfigs"][0]["natIP"]
-    print(external_ip)
-
     ssh = paramiko.SSHClient()
     ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 
@@ -53,6 +51,7 @@ def start_runner(
     max_wait_time = 600
     wait_interval = 10
     waited_time = 0
+    response = None
 
     while waited_time < max_wait_time:
         response = (
@@ -61,7 +60,6 @@ def start_runner(
         status = response.get("status")
 
         if status == "RUNNING":
-            print("Yaaaay")
             break
 
         time.sleep(wait_interval)
@@ -76,7 +74,7 @@ def start_runner(
 
 def stop_runner(creds):
     compute = authenticate_vm(creds)
-    request = compute.instances().start(
+    request = compute.instances().stop(
         project="gpu-insatnce", zone="us-central1-a", instance="demos-tests"
     )
     request.execute()

diff --git a/.github/workflows/gcp-test.yaml b/.github/workflows/gcp-test.yaml
@@ -32,7 +32,7 @@ jobs:
            echo "${{ secrets.SSH_KEY }}" > $HOME/.ssh/id_rsa
            chmod 600 $HOME/.ssh/id_rsa
            python3 db_auth.py ${{ secrets.DB_ENDPOINT }} ${{ secrets.DB_OBJ_ID }}
-           python3 vm_auth.py ${{ secrets.SSH_USERNAME }} ${{ secrets.SSH_PASSPHRASE }} "true"
+           python3 vm_auth.py ${{ secrets.SSH_USERNAME }} ${{ secrets.SSH_PASSPHRASE }} "false"
 
   run-test:
     needs: activate-vm
@@ -71,6 +71,6 @@ jobs:
           run: |
             cd demos/.github/auth
             python3 db_auth.py ${{ secrets.DB_ENDPOINT }} ${{ secrets.DB_OBJ_ID }}
-            python3 vm_auth.py ${{ secrets.SSH_USERNAME }} ${{ secrets.SSH_PASSPHRASE }} "false"
+            python3 vm_auth.py ${{ secrets.SSH_USERNAME }} ${{ secrets.SSH_PASSPHRASE }} "true"