The current code base only has the augmentations as “classes”, but we also need to give the option to have them as functions `fn add_noise(args)`