Issue with Deep-Reinforcement Learning Part 2 Code

[KobbyObeng]

Dear Rahul,
Upon following your tutorial with respect to the REINFORCE algorithm, the training aspect of the code gives me an error which I cannot resolve. Kindly find the error pasted in the below. I would be grateful if you could help me out.

(https://github.com/Rahul-Choudhary-3614/Deep-Reinforcement-Learning-Notebooks/assets/148027862/a5a6a089-a664-4da4-8c4d-2b3c21d90779)