pipeline {
    agent {
        dockerfile {
			 additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} --build-arg CUTOFF=${params.CUTOFF} -t sebastianwalesa"
        }
    }
    parameters {
        string(
            defaultValue: '1000',
            description: 'Number of epochs',
            name: 'EPOCHS',
            trim: false
        )
    }
    stages {
        stage('MLflow-stage'){
		steps {
                copyArtifacts filter: '*', projectName: 's478839-create-dataset'
			    sh 'python3 ml_pytorch_mlflow.py with "epochs=$EPOCHS"'
		}
	}
	}

}