node {
    checkout scm
    def testImage = docker.build("test-image") 
    testImage.inside {
        sh 'python3 create_dataset.py 6'
    }
}