pipeline {
    agent {
        dockerfile true 
    }

    stages {
        stage("Checkout"){
            steps{
                git "https://git.wmi.amu.edu.pl/s151636/ium_151636.git"
            }
        }

        stage("DVC"){
            steps{
                withCredentials(
                [sshUserPrivateKey(credentialsId: '48ac7004-216e-4260-abba-1fe5db753e18', keyFileVariable: 'IUM_SFTP_KEY', passphraseVariable: '', usernameVariable: '')]) {
                sh 'dvc remote add -d ium_ssh_remote ssh://ium-sftp@tzietkiewicz.vm.wmi.amu.edu.pl/ium-sftp'
                sh 'dvc remote modify --local ium_ssh_remote keyfile $IUM_SFTP_KEY'
                sh 'dvc pull'
                sh 'dvc repro'}
            }
        }
        stage('Archive Output') {
            steps {
                archiveArtifacts 'model.pth'
            }
        }
    }
}