diff --git a/requirements.txt b/requirements.txt index 7abcb54a..8f7c30c0 100644 --- 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 index c33c2154..69c5fa4b 100644 --- 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: