Dockerfile

Jenkinsfile

@@ -1,10 +1,5 @@
 pipeline {
-  agent { 
+  agent any
-    dockerfile {
-      filename 'Dockerfile'
-      args '-u root'
-    }
-  }
   
   parameters {
     string (
@@ -21,17 +16,29 @@ pipeline {
   }
 
   stages {
-    stage('Run create-dataset script') {
+    stage('Download dataset') {
       steps {
-        withEnv (["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
+        withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
           sh 'kaggle datasets download -d mlg-ulb/creditcardfraud'
           sh 'unzip creditcardfraud.zip'
           sh 'rm creditcardfraud.zip'
+        }
+      }
+    }
+
+    stage('Run create-dataset script') {
+        agent {
+          dockerfile {
+            reuseNode true
+        }
+      }
+
+      steps {
         sh 'chmod +x create-dataset.py'
         sh 'python3 ./create-dataset.py $KAGGLE_USERNAME'
       }
     }
-    }
+
     stage('Archive Artifacts') {
       steps {
         archiveArtifacts artifacts: 'data/*', onlyIfSuccessful: true