Merge pull request #247 from runpod/skypilot

Skypilot
runpod · Dec 4, 2023 · 27d8401 · 27d8401
2 parents e4f7043 + 48b7880
commit 27d8401
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ aiohttp-retry >= 2.8.3
 backoff >= 2.2.1
 boto3 >= 1.26.165
 click >= 8.1.7
-colorama >= 0.4.6
+colorama >= 0.2.5, < 0.4.5
 fastapi[all] >= 0.94.0
 paramiko >= 3.3.1
 prettytable >= 3.9.0

diff --git a/runpod/cli/groups/ssh/functions.py b/runpod/cli/groups/ssh/functions.py
@@ -30,7 +30,7 @@ def get_user_pub_keys():
     Get the current user's SSH keys
     '''
     user = get_user()
-    keys = user['pubKey']
+    keys = '' if user['pubKey'] is None else user['pubKey']
 
     # Parse the keys
     keys = keys.split('\n')
@@ -86,7 +86,7 @@ def add_ssh_key(public_key):
     Checks if the key already exists before adding it.
     """
     user = get_user()
-    current_keys = user['pubKey']
+    current_keys = '' if user['pubKey'] is None else user['pubKey']
 
     # Check if the key already exists
     if public_key in current_keys: