From d20cec2c0700fdee848fd277f05665dd3256f9ea Mon Sep 17 00:00:00 2001 From: Codechickdev Date: Wed, 28 Dec 2022 15:00:02 +0530 Subject: [PATCH] Solved Issues #8 :hammer: :heavy_check_mark: --- .gitignore | 4 ++++ opendatasets/utils/kaggle_api.py | 18 +++++++++++------- 2 files changed, 15 insertions(+), 7 deletions(-) diff --git a/.gitignore b/.gitignore index b6e4761..3cea6b6 100644 --- a/.gitignore +++ b/.gitignore @@ -127,3 +127,7 @@ dmypy.json # Pyre type checker .pyre/ + +# Production Folders and Files for testing +production/ +production/* diff --git a/opendatasets/utils/kaggle_api.py b/opendatasets/utils/kaggle_api.py index 82361c8..f139652 100644 --- a/opendatasets/utils/kaggle_api.py +++ b/opendatasets/utils/kaggle_api.py @@ -18,15 +18,19 @@ def _get_kaggle_key(): def read_kaggle_creds(): + path: str = None try: if os.path.exists('./kaggle.json'): - with open('./kaggle.json', 'r') as f: - key = f.read() - data = json.loads(key) - if 'username' in data and 'key' in data: - os.environ['KAGGLE_USERNAME'] = data['username'] - os.environ['KAGGLE_KEY'] = data['key'] - return True + path = './kaggle.json' + elif os.path.exists('../root/.kaggle/kaggle.json'): + path = '../root/.kaggle/kaggle.json' + with open(path, 'r') as f: + key = f.read() + data = json.loads(key) + if 'username' in data and 'key' in data: + os.environ['KAGGLE_USERNAME'] = data['username'] + os.environ['KAGGLE_KEY'] = data['key'] + return True except Exception: return False