fix

Author: lperron

examples/python/prize_collecting_tsp.py

@@ -14,6 +14,7 @@
 
 """Simple prize collecting TSP problem with a max distance."""
 
+from ortools.routing import enums_pb2
 from ortools.routing.python import routing
 
 
@@ -145,10 +146,10 @@ def distance_callback(from_index, to_index):
     # Setting first solution heuristic.
     search_parameters = routing.default_routing_search_parameters()
     search_parameters.first_solution_strategy = (
-        routing.FirstSolutionStrategy.PATH_CHEAPEST_ARC
+        enums_pb2.FirstSolutionStrategy.PATH_CHEAPEST_ARC
     )
     search_parameters.local_search_metaheuristic = (
-        routing.LocalSearchMetaheuristic.GUIDED_LOCAL_SEARCH
+        enums_pb2.LocalSearchMetaheuristic.GUIDED_LOCAL_SEARCH
     )
     search_parameters.time_limit.seconds = 15
     # search_parameters.log_search = True

examples/python/prize_collecting_vrp.py

@@ -14,6 +14,7 @@
 
 """Simple prize collecting VRP problem with a max distance."""
 
+from ortools.routing import enums_pb2
 from ortools.routing.python import routing
 
 
@@ -155,10 +156,10 @@ def distance_callback(from_index, to_index):
     # Setting first solution heuristic.
     search_parameters = routing.default_routing_search_parameters()
     search_parameters.first_solution_strategy = (
-        routing.FirstSolutionStrategy.PATH_CHEAPEST_ARC
+        enums_pb2.FirstSolutionStrategy.PATH_CHEAPEST_ARC
     )
     search_parameters.local_search_metaheuristic = (
-        routing.LocalSearchMetaheuristic.GUIDED_LOCAL_SEARCH
+        enums_pb2.LocalSearchMetaheuristic.GUIDED_LOCAL_SEARCH
     )
     search_parameters.time_limit.seconds = 15
     # search_parameters.log_search = True