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