pipeline {
    agent {
		docker { image 'kamilguttmann/ium:train' }
	}
    parameters {
        string (
                defaultValue: '100',
                description: 'Epochs',
                name: 'EPOCHS'
            )
        buildSelector(
            defaultSelector: lastSuccessful(),
            description: 'Which build to use for copying artifacts',
            name: 'BUILD_SELECTOR'
        )
    }
    stages {
        stage('Copy artifacts') {
            steps {
                copyArtifacts fingerprintArtifacts: true, projectName: 's444380-create-dataset', selector: buildParameter('BUILD_SELECTOR')
            }
        }
        stage("Checkout git") {
            steps {
	        checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's444380', url: 'https://git.wmi.amu.edu.pl/s444380/ium_444380.git']]])
            }
        }
        stage("Train model") {
            steps {
                sh "chmod u+x ./train_model.py"
                sh "python3 ./train_model.py $EPOCHS"
                archiveArtifacts artifacts: "model/*, out.csv", onlyIfSuccessful: true
            }
        }
    }
    post {
        success {
            emailext body: "SUCCESS", subject: "s444380-training", to: "e19191c5.uam.onmicrosoft.com@emea.teams.ms"
            build job: "s444380-evaluation/master"
        }

        failure {
            emailext body: "FAILURE", subject: "s444380-training", to: "e19191c5.uam.onmicrosoft.com@emea.teams.ms"
        }
    }
}