pipeline {
    agent {
        docker { image 's444507_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: 's444507-create-dataset', selector: lastSuccessful()
            }
        }
        stage('Train model with sacred') {
            steps {
                sh "python3 IUM_07/deepLearning_sacred.py $epoch"
            }
        }
    }
    post {
        success {
            archiveArtifacts artifacts: 'prediction_results.csv, CarPrices_pytorch_model.pkl', followSymlinks: false
        }
        always {
            emailext body: "${currentBuild.currentResult}", subject: 's444507-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
        }
    }
}