I have been trying to use TF-Agents, but it is so stuck up on the Training error, I have no idea how to resolve this,

I either have broadcast error, or I have this,
python3.10/site-packages/tensorflow/python/ops/check_ops.py", line 1153, in assert_rank
raise ValueError(
ValueError: Tensor must have rank 3. Received rank 2, shape (1, 0)

fixing one produces the other in a loop. I am finally giving up on TF-Agent and moving to some other frame work.