Update dataloader.py

dataloader.py

@@ -20,7 +20,7 @@ class VRP_Dataset(Dataset):
             raise ValueError("Not enough rows in CSV to build required dataset")
 
         # Randomly sample and reshape
-        coords = torch.tensor(raw_data[['longitude', 'latitude']].values[:dataset_size * num_nodes], dtype=torch.float32)
+        coords = torch.tensor(raw_data[['lng', 'lat']].values[:dataset_size * num_nodes], dtype=torch.float32)
         node_positions = coords.view(dataset_size, num_nodes, 2)
         self.node_positions = node_positions