Update dataset.py

Minor doc update.