pipeline {
    agent {
        docker {
            image 's444409-create-dataset'
        }
    }
    
    stages {
        stage('Train model') {
            steps {
                sh "python train_model.py"
            }
        }
    }

    post {
        always {
            archiveArtifacts artifacts: 'model_out', 
            onlyIfSuccessful: true
        }
    }
}