Update scrub.py

scrub.py

@@ -57,7 +57,7 @@ reversed_map = {
 
 class CustomDataset(Dataset):
     
-    def __init__(self, forget_class=16, root, transformations = None):
+    def __init__(self,root,forget_class=16, transformations = None):
         
         self.transformations = transformations
         self.im_paths = [im_path for im_path in sorted(glob(f"{root}/*/*"))]
@@ -111,7 +111,7 @@ class SingleCelebCustomDataset(Dataset):
         return im, gt
 
     
-def get_dls(root, forget_class=16, transformations, bs, split = [0.9, 0.05, 0.05], ns = 4, single=False):
+def get_dls(root, transformations, bs, forget_class=16, split = [0.9, 0.05, 0.05], ns = 4, single=False):
     if single:
         ds = SingleCelebCustomDataset(root = root, forget_class=forget_class, transformations = transformations)
     else: