pipeline {
    agent {
        docker { image 'adnovac/ium_s434760:1.0' }
    }
    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: 's434760-create-dataset', selector: buildParameter('WHICH_BUILD'))
                }
            }
            stage('train')
            {
                steps
                {
                    catchError {
                        sh 'python3.8 train.py ${BATCH_SIZE} ${EPOCHS}'
                    }
                }
            }
            stage('archive artifacts') {
                steps {
                    archiveArtifacts 'model.h5'
                }
            }
            stage('send email') {
                steps {
                    emailext body: currentBuild.result ?: 'SUCCESS',
                    subject: 's434760 - train',
                    to: 'annnow19@st.amu.edu.pl'
                }
            }
    }
}