pipeline {
    agent {
        dockerfile true
    }
    parameters {
        string(
            defaultValue: "10",
            description: 'Number of epochs in training',
            name: 'NO_OF_EPOCHS'
        )
    }
    stages {
        stage("Git clone") {
            steps {
              checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's444465', url: 'https://git.wmi.amu.edu.pl/s444465/ium_444465']]])
            }
        }
        stage("Run 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'
                    }
            }
        }
    }
}