Can't measure the fidelity of the states.

I have measure the quantum states using tfq.layers.State(). The output of tfq.layers.State() is of the form tf.RaggedTensor. Now I have measured two different states, using the same method. (say state1 and state2)
After using cirq.fidelity(state1, state2), I am getting the following error:

ValueError: Exception encountered when calling layer "generator__discriminator_6" (type Generator_Discriminator).

Failed to infer the qid shape of the given states. Please specify the qid shape explicitly using the `qid_shape` argument.

Call arguments received:
  • inputs=tf.Tensor(shape=(1,), dtype=string)

Why is it so? And how can I measure the fidelity of the states using tfq.layers.State()?

Start with the reported tfq.layers.State() output and the cirq.fidelity(state1, state2) call, then trace how the resulting tf.RaggedTensor is passed into fidelity. Check the relevant TensorFlow Quantum and Cirq API behavior for qid_shape handling. Done means the cause of the error is documented and a valid way to measure fidelity from two State() results is demonstrated.