Update dataset.py

[idiomaticrefactoring]

refactoring code with var unpack which is more pythonic, concise, readable and efficient; how do think this change which has practical value？