tf agent: Time Step Spec Mismatch

Question

Any know what\'s wrong or how to fix this?

ValueError: Given time_step: TimeStep(step_type=array(0), reward=array(0., dtype=float32), discount=array(1., dtype=float32