Convert demand to list

duducheng · Jun 22, 2018 · 5117e74 · 5117e74
1 parent 1935bea
commit 5117e74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generate_data.py b/generate_data.py
@@ -18,7 +18,7 @@ def generate_vrp_data(dataset_size, vrp_size):
     return list(zip(
         np.random.uniform(size=(dataset_size, 2)).tolist(),  # Depot location
         np.random.uniform(size=(dataset_size, vrp_size, 2)).tolist(),  # Node locations
-        np.random.randint(1, 10, size=(dataset_size, vrp_size)),  # Demand, uniform integer 1 ... 9
+        np.random.randint(1, 10, size=(dataset_size, vrp_size)).tolist(),  # Demand, uniform integer 1 ... 9
         np.ones(dataset_size) * CAPACITIES[vrp_size]  # Capacity, same for whole dataset
     ))