from torch.utils.data import DataLoader

from bevy_zeroverse_dataloader import ChunkedDataset

from test import visualize


def main():
    chunked_dataset = ChunkedDataset("data/zeroverse/cli")
    dataloader = DataLoader(chunked_dataset, batch_size=1, shuffle=False)

    for batch in dataloader:
        visualize(batch)


if __name__ == "__main__":
    main()