pipeline {
  	agent {
		dockerfile {
			 additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=51854c15661583860a60db232a7026f0 --build-arg CUTOFF=${params.CUTOFF} -t kgirejko"
		}
	  }
     parameters {
		string(
			defaultValue: 'kgirejko',
			description: 'Kaggle username',
            name: 'KAGGLE_USERNAME',
            trim: false
		)

		password(
			defaultValue: '51854c15661583860a60db232a7026f0',
            description: 'Kaggle token taken from kaggle.json file, as described in https://github.com/Kaggle/kaggle-api#api-credentials',
            name: 'KAGGLE_KEY'
		)
		
		string(
            defaultValue: '1000',
            description: 'Number of epochs',
            name: 'EPOCHS',
            trim: false
        )
     }

   stages {
	stage('Main train'){
		steps {
            copyArtifacts filter: '*', projectName: 's478815-create-dataset'
			sh 'python3 ./train-sacred.py with "epochs=$EPOCHS"'
			
			archiveArtifacts artifacts: 'experiment/_sources/*, experiment/**, modelS.pkl, output.txt'
			sh 'rm -r experiment'
			
			build job: 's478815-evaluation/master/'
		}
	}
    }

}