minor changes in cvrp/local_search.py

ai4co · Feb 14, 2025 · 9261d20 · 9261d20
1 parent d4695c3
commit 9261d20
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/rl4co/envs/routing/cvrp/local_search.py b/rl4co/envs/routing/cvrp/local_search.py
@@ -35,7 +35,9 @@
 assert os.path.isfile(HGS_LIBRARY_FILEPATH)  # Failure case will be handled in the env.py
 
 
-def local_search(td: TensorDict, actions: torch.Tensor, max_iterations: int = 1000) -> torch.Tensor:
+def local_search(
+    td: TensorDict, actions: torch.Tensor, max_iterations: int = 1000, **kwargs
+) -> torch.Tensor:
     """
     Improve the solution using local search for CVRP, based on PyVRP.
 
@@ -365,7 +367,7 @@ def local_search(self, data, routes: List[np.ndarray], count: int = 1) -> List[n
         assert dist_mtx.shape[0] == dist_mtx.shape[1]
         assert (dist_mtx >= 0.0).all()
 
-        callid = (time.time_ns()*10000+random.randint(0,10000))%C_INT_MAX
+        callid = (time.time_ns()*100000+random.randint(0,100000))%C_INT_MAX
 
         tmppath = "/tmp/route-{}".format(callid)
         resultpath = "/tmp/swapstar-result-{}".format(callid)