pipeline {
    agent {
        dockerfile {
			 additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} --build-arg CUTOFF=${params.CUTOFF} -t sebastianwalesa"
        }
    }
    parameters {
        string(
            defaultValue: '1000',
            description: 'Number of epochs',
            name: 'EPOCHS',
            trim: false
        )
    }
    stages {
        stage('Train-stage'){
		steps {
                copyArtifacts filter: '*', projectName: 's478839-create-dataset'
			    sh 'python3 ./ml_pytorch.py $EPOCHS'
			    archiveArtifacts artifacts: 'Model_xPosition.pkl', followSymlinks: false
		}
	}
    }
    post {
        success {
            emailext body: 'SUCCESS', subject: 's478839-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
        failure {
            emailext body: 'FAILURE', subject: 's478839-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
        unstable {
            emailext body: 'UNSTABLE', subject: 's478839-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
        changed {
            emailext body: 'CHANGED', subject: 's478839-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }		
    }

}