add imports

Author: Dnafivuq

task_3/taskdata.py

@@ -1,3 +1,8 @@
+from typing import Tuple
+import torch
+from torch.utils.data import Dataset
+
+
 class TaskDataset(Dataset):
     def __init__(self, transform=None):