Update dataset.py

Fixed incomplete docstring.