Change import DataLoader from torch_geometric.data to torch_geometric.loader to avoid a warning message