data module must be class based with the following requirements: attributes: - path to data - number of batch size - number of workers method: - dataset builder - dataloader builder