pipeline {
    agent {
        docker { image 's444354_create_dataset_image:latest' }
    }
    parameters {
        string(name: 'epoch', defaultValue: '1000', description: 'Number of epochs to train model.')
    }
    stages {
        stage('Get arifacts') {
            steps {
                copyArtifacts fingerprintArtifacts: true, projectName: 's444354-create-dataset', selector: lastSuccessful()
            }
        }
        stage('Show stats') {
            steps {
                sh "python3 ./pytorch/pytorch.py $epoch"
            }
        }
    }
    post {
        success {
            archiveArtifacts artifacts: 'pytorch/result.txt', followSymlinks: false
        }
        always {
            emailext body: "${currentBuild.currentResult}", subject: 's444354-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
    }
}