Wrong return value in train_test_data function

[ham45X13y]

In the pytorch_mebeauty_dataset.py file in the function train_test_data() there is a mix up with test and training data. There are two issues. One is the testloader data that is provided and the other is the return value that is provided. This is the current code:

train_data = MEBeauty(root_dir, train_scores,test_scores, train = True, transform = transform_train)
    trainloader = torch.utils.data.DataLoader(train_data, batch_size = batch, shuffle =True, num_workers = num_workers, pin_memory = pin_memory)
    test_data = MEBeauty(root_dir, train_scores,test_scores, train = False, transform = transform_test)
    testloader = torch.utils.data.DataLoader(train_data , batch_size = batch, shuffle =True, num_workers = num_workers, pin_memory = pin_memory)

    return trainloader, trainloader

That I assume should be:

train_data = MEBeauty(root_dir, train_scores,test_scores, train = True, transform = transform_train)
    trainloader = torch.utils.data.DataLoader(train_data, batch_size = batch, shuffle =True, num_workers = num_workers, pin_memory = pin_memory)
    test_data = MEBeauty(root_dir, train_scores,test_scores, train = False, transform = transform_test)
    testloader = torch.utils.data.DataLoader(test_data, batch_size = batch, shuffle =True, num_workers = num_workers, pin_memory = pin_memory)

    return trainloader, testloader

Can someone verfiy if this is an actual issue or if this is done on purpose?