pipeline {
  	agent {
		docker { image 'korneliag/ium478815:1.0' }
		}
	  }
     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-mlflow.py with "epochs=$EPOCHS"'

		}
	}
    }

}