Set docker image to python 3.6