pipeline {
    agent {docker { image 'snowycocoon/ium_434788:4'}}
	parameters{
        buildSelector(
            defaultSelector: lastSuccessful(),
            description: 'Which build to use for copying artifacts',
            name: 'WHICH_BUILD'
        )
        string(
            defaultValue: '16',
            description: 'batch size',
            name: 'BATCH_SIZE'
        )
        string(
            defaultValue: '15', 
            description: 'epochs',
            name: 'EPOCHS'
        )
    }

    stages {
		stage('copy artifacts')
            {
                steps
                {
                    copyArtifacts(fingerprintArtifacts: true, projectName: 's434788-create-dataset', selector: buildParameter('WHICH_BUILD'))
                }
            }
            stage('train')
            {
                steps
                {
                    catchError {
                        sh 'python3.8 Zadanie_06_and_07_training.py ${BATCH_SIZE} ${EPOCHS}'
                    }
                }
            }

		stage('Archive artifacts') {
			steps{
				archiveArtifacts 'wine_model.h5'
                archiveArtifacts 'my_runs/**'
			}
		}
	}

    post {
        success {

            build job: 's434788-evaluation/master'
            mail body: 'SUCCESS',
                subject: 's434788 training',
                to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }
        unstable {
            mail body: 'UNSTABLE', subject: 's434788 training', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }
        failure {
            mail body: 'FAILURE', subject: 's434788 training', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }
        changed {
            mail body: 'CHANGED', subject: 's434788 training', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }
    }
}