Correct minor error in docstring for hk.Embed.
Correct minor error in docstring for hk.Embed.

The docstring used to refer to a function argument using an incorrect name.
This is now fixed.

PiperOrigin-RevId: 345686155
Change-Id: Ibe9f4e46b9b61f50e2732a9f032d8f0b29fed92c

deepmind/dm-haiku,deepmind/dm-haiku