Use correct dataset type in `from_generator` docs
_The documentation is not available anymore as the PR was closed or merged._
Use the correct dataset type in the `from_generator` docs (example with sharding).