Dataset and DataLoader Example - Pytorch Typing CST Test