import tensorflow_datasets as tfds


def load_nyu():
    builder = tfds.builder('nyu_depth_v2')
    builder.download_and_prepare(download_dir='../nyu')
    return builder.as_dataset(split='train', shuffle_files=True)


def print_hi(name):
    # Use a breakpoint in the code line below to debug your script.
    print(f'Hi, {name}')  # Press Ctrl+F8 to toggle the breakpoint.


# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    load_nyu()

# See PyCharm help at https://www.jetbrains.com/help/pycharm/