chore(docs): fix typos in docstrings

[jeertmans]

Hello!

This is a small PR to fix typos in docstrings.

Maybe, I would suggest adding an import for dataclass in the example (otherwise it will not run), and maybe indicate that it works with other dataclass decorators, like the dataclass decorator from chex.

[patrick-kidger]

Thank you for the fixes! These look great to me. :)

Good points about the import and the other dataclass options. I'll tweak the docs for this one now.

[jeertmans]

As a suggestion, including doctests inside your test workflow might help to catch that kind of "errors" in the future :-)

Pytest includes support for that, which you can automatically enable in your pyproject.toml file:

[tool.pytest.ini_options]
addopts = [
  '--doctest-modules',
]

[patrick-kidger]

Hmm, I do often use non-runnable incomplete code snippets in docs, so I'm not completely sold. Happy to consider a PR on this if it doesn't change too much, but to be honest I don't feel so strongly that I'll add it myself ;)