node {
    checkout scm
    docker.image('s444452/ium:1.3').inside {
        stage('Preparation') {
            properties([
                parameters([
                string(
                 defaultValue: ".",
                 description: 'Arguments for model training: arg1,arg2,arg3',
                 name: 'TRAIN_ARGS'
                )
                ])
            ])
        }
        stage('Run script') {
            withEnv(["TRAIN_ARGS=${params.TRAIN_ARGS}"]) {
                sh "python3 Scripts/train_neural_network.py $TRAIN_ARGS"
            }
        }
    }
}