pipeline {
   agent {
      dockerfile true
   }
   parameters{
      buildSelector(
            defaultSelector: lastSuccessful(),
            description: 'Which build to use for copying artifacts',
            name: 'WHICH_BUILD'
        )
        string(
            defaultValue: '64',
            description: 'batch size',
            name: 'BATCH_SIZE'
        )
        string(
            defaultValue: '100',
            description: 'epochs',
            name: 'EPOCHS'

        )
   }
   stages {
      stage('checkout') {
            steps {
               copyArtifacts fingerprintArtifacts: true, projectName: 's434732-create-dataset', selector: buildParameter('WHICH_BUILD')
            }
        }
      stage('Docker'){
         steps{
            sh 'python3 "./training.py" ${BATCH_SIZE} ${EPOCHS} > model_pred.txt'
         }
      }
      stage('archiveArtifacts') {
			steps{
                archiveArtifacts 'ium_s434732/**'
				archiveArtifacts 'model_pred.txt'
                archiveArtifacts 'FootballModel.pth'
			}
		}
    }
    post {
        success {
            build job: 's434732-evaluation/master'
            mail body: 'SUCCESS TRAINING', subject: 's434766', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }

        failure {
            mail body: 'FAILURE TRAINING', subject: 's434732', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
        }

   }
}