pipeline {
    agent {
        docker { image 'sebastianwalesa/ium:1.0' }
    }
    parameters {
        string(
            defaultValue: '1000',
            description: 'Number of epochs',
            name: 'EPOCHS',
            trim: false
        )
    }
    stages {
        stage('MLflow-stage'){
		steps {
                copyArtifacts filter: '*', projectName: 's478839-create-dataset'
			    sh 'python3 ml_pytroch_mlflow.py with "epochs=$EPOCHS"'
		}
	}
	}

}