Update src/train_transformer.py

src/train_transformer.py

@@ -16,7 +16,7 @@ LR = 1e-3
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # Dataset wrapper
-dclass EmotionDataset(Dataset):
+class EmotionDataset(Dataset):
     def __init__(self, X, y):
         self.X = torch.tensor(X, dtype=torch.long)
         self.y = torch.tensor(y, dtype=torch.long)