pipeline{
    agent any

    parameters {
        gitParameter(
            name: 'BRANCH',
            type: 'branch',
            defaultValue: 'master',
            description: 'Select the branch of training'
        )
    }

    stages{
         stage('Copy model artifacts with test dataset') {
            steps {
                copyArtifacts(
                    projectName: 'z-s487175-training',
                    fingerprintArtifacts: true,
                    selector: [$class: 'SpecificBuildSelector', buildNumber: "lastSuccessfulBuild"]
                )
            }
        }

        stage('Train model'){
            steps{
                sh "docker run -v ${env.WORKSPACE}/dane:/app ium python3 /app/DL-prediction.py"
            }
        }

        stage('Archive prediction results'){
            steps{
                sh "docker cp \$(docker ps -l -q):/app/results_prediction.csv ${env.WORKSPACE}"
                archiveArtifacts artifacts: 'results_prediction.csv', fingerprint: true
            }
        }
    }
}