I have cloned this github and am trying to execute the code using tiny dataset. The error messages shows no module named 'zsh' which is used for data sampling i guess. Can you provide instruction to run the code? and the requirements? Thanks