Question: What dataset do they use?
Answer: A parallel corpus where the source is an English expression of code and the target is Python code.
Alternative answer:  text-code parallel corpus