diff --git a/Jenkinsfile b/Jenkinsfile index 58049c5..81562f1 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -23,9 +23,9 @@ pipeline { stages { stage('Run create-dataset script') { steps { - withEnv (["KAGGLE_USERNAME=${env.KAGGLE_USERNAME}", "KAGGLE_KEY=${env.KAGGLE_KEY}"]) { + withEnv (["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}"]) { sh 'chmod +x create-dataset.py' - sh 'python3 ./create-dataset.py $KAGGLE_KEY' + sh 'python3 ./create-dataset.py $KAGGLE_USERNAME' } } } diff --git a/create-dataset.py b/create-dataset.py index 891b368..209d2ac 100644 --- a/create-dataset.py +++ b/create-dataset.py @@ -11,8 +11,8 @@ from sklearn.model_selection import train_test_split def download_kaggle_dataset(): print(sys.argv) - os.environ["KAGGLE_USERNAME"] = sys.argv[1] - os.environ["KAGGLE_KEY"] = sys.argv[2] + os.environ["KAGGLE_USERNAME"] = "vskyper" + os.environ["KAGGLE_KEY"] = sys.argv[1] kaggle = KaggleApi() kaggle.authenticate() kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True)