diff --git a/tests/utils/forge.py b/tests/utils/forge.py index d364ef1..db29572 100644 --- a/tests/utils/forge.py +++ b/tests/utils/forge.py @@ -7,8 +7,12 @@ from github import Github def read_token(): token_file = Path(os.getcwd(), 'tests', 'token.txt') - with open(token_file, 'r') as f: - token = f.read() + if token_file.exists(): + with open(token_file, 'r') as f: + token = f.read() + + else: + token = os.environ.get('token') return token