Update dataset_dict.py

shape returns (number of rows, number of columns)