diff --git a/create-dataset.py b/create-dataset.py index abe9649..2021e72 100644 --- a/create-dataset.py +++ b/create-dataset.py @@ -10,8 +10,7 @@ from sklearn.model_selection import train_test_split def download_kaggle_dataset(): - os.environ["KAGGLE_USERNAME"] = sys.argv[1] - os.environ["KAGGLE_KEY"] = sys.argv[2] + print(os.environ.get("KAGGLE_USERNAME")) kaggle = KaggleApi() kaggle.authenticate() kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True)