Thanks to Thomas Simonini for finessing it😎
https://github.com/huggingface/deep-rl-class/blob/main/unit4/unit4.ipynbÂ