pipeline {
    agent any
    parameters {
        buildSelector(
            name: 'BUILD_SELECTOR',
            defaultSelector: lastSuccessful(),
            description: 'A build to take the artifacts from'
        )
        string(
            name: 'EPOCHS',
            description: 'Number of epochs',
            defaultValue: '10'
        )
    }
    stages {
        stage('Copy artifacts') {
            steps {
                script {
                    copyArtifacts(
                        projectName: 'z-s487184-create-dataset',
                        selector: buildParameter('BUILD_SELECTOR'),
                        target: './'
                    )
                }
            }
        }
        stage('Run training and save model') {
            steps {
                    script {
                    sh 'ls -l'
                    docker.image('docker-image').inside {
                            sh 'ls -l'
                            sh 'python3 ./train.py'
                            archiveArtifacts 'model.pt'
                    }
                }
            }
        }
    }
    post {
        success {
            build job: 'z-s487184-evaluation.eg/main', wait: false
        }
    }
}