I was following this tutorial (https://www.tensorflow.org/agents/tutorials/1_dqn_tutorial?hl=en) on how to implement a Deep Q Network algorithm with TF Agents to solve the C