Correcting a typo I made by mistake in last cl before submitting, sorry about that.
Correcting a typo I made by mistake in last cl before submitting, sorry about that.

PiperOrigin-RevId: 441586918
Change-Id: I7ebbe0cdbb0013c56c4a78e83d12836c448a3cee

google-research/circuit_training